SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Update SwinForMaskedImageModeling doctest values (#20139)

Browse Source 
* Update doctest values

* Update copy statement
This commit is contained in:
amyeroberts
2022-11-09 14:53:01 +00:00
committed by GitHub
parent c4cad8e301
commit d606d566ab
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/swin/modeling_swin.py
View File

@@ -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

2
src/transformers/models/swinv2/modeling_swinv2.py
View File

@@ -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)