Unverified Commit d606d566 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Update SwinForMaskedImageModeling doctest values (#20139)

* Update doctest values

* Update copy statement
parent c4cad8e3
...@@ -1072,7 +1072,7 @@ class SwinForMaskedImageModeling(SwinPreTrainedModel): ...@@ -1072,7 +1072,7 @@ class SwinForMaskedImageModeling(SwinPreTrainedModel):
>>> outputs = model(pixel_values, bool_masked_pos=bool_masked_pos) >>> outputs = model(pixel_values, bool_masked_pos=bool_masked_pos)
>>> loss, reconstructed_pixel_values = outputs.loss, outputs.logits >>> loss, reconstructed_pixel_values = outputs.loss, outputs.logits
>>> list(reconstructed_pixel_values.shape) >>> list(reconstructed_pixel_values.shape)
[1, 3, 224, 224] [1, 3, 192, 192]
```""" ```"""
return_dict = return_dict if return_dict is not None else self.config.use_return_dict return_dict = return_dict if return_dict is not None else self.config.use_return_dict
......
...@@ -1099,7 +1099,7 @@ class Swinv2Model(Swinv2PreTrainedModel): ...@@ -1099,7 +1099,7 @@ class Swinv2Model(Swinv2PreTrainedModel):
""", """,
SWINV2_START_DOCSTRING, SWINV2_START_DOCSTRING,
) )
# Copied from transformers.models.swin.modeling_swin.SwinForMaskedImageModeling with swin->swinv2, base-simmim-window6-192->tiny-patch4-window8-256, SWIN->SWINV2,Swin->Swinv2, 224->256 # Copied from transformers.models.swin.modeling_swin.SwinForMaskedImageModeling with swin->swinv2, base-simmim-window6-192->tiny-patch4-window8-256,SWIN->SWINV2,Swin->Swinv2,192->256
class Swinv2ForMaskedImageModeling(Swinv2PreTrainedModel): class Swinv2ForMaskedImageModeling(Swinv2PreTrainedModel):
def __init__(self, config): def __init__(self, config):
super().__init__(config) super().__init__(config)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment