From d606d566ab0a2635718cb5210d1dad8fae4ce112 Mon Sep 17 00:00:00 2001 From: amyeroberts <22614925+amyeroberts@users.noreply.github.com> Date: Wed, 9 Nov 2022 14:53:01 +0000 Subject: [PATCH] Update SwinForMaskedImageModeling doctest values (#20139) * Update doctest values * Update copy statement --- src/transformers/models/swin/modeling_swin.py | 2 +- src/transformers/models/swinv2/modeling_swinv2.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/transformers/models/swin/modeling_swin.py b/src/transformers/models/swin/modeling_swin.py index c64f416fe2..21ee53481a 100644 --- a/src/transformers/models/swin/modeling_swin.py +++ b/src/transformers/models/swin/modeling_swin.py @@ -1072,7 +1072,7 @@ class SwinForMaskedImageModeling(SwinPreTrainedModel): >>> outputs = model(pixel_values, bool_masked_pos=bool_masked_pos) >>> loss, reconstructed_pixel_values = outputs.loss, outputs.logits >>> 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 diff --git a/src/transformers/models/swinv2/modeling_swinv2.py b/src/transformers/models/swinv2/modeling_swinv2.py index 91b9f38337..95eba10bd0 100644 --- a/src/transformers/models/swinv2/modeling_swinv2.py +++ b/src/transformers/models/swinv2/modeling_swinv2.py @@ -1099,7 +1099,7 @@ class Swinv2Model(Swinv2PreTrainedModel): """, 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): def __init__(self, config): super().__init__(config)