From 1ff8d3e5bbeddbf8f35389ca225c9e0695eb7173 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Wed, 15 Mar 2023 17:15:40 +0100 Subject: [PATCH 1/2] fix --- src/transformers/models/vit/modeling_vit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/vit/modeling_vit.py b/src/transformers/models/vit/modeling_vit.py index 154afdb211f3..3bfdefb08b66 100644 --- a/src/transformers/models/vit/modeling_vit.py +++ b/src/transformers/models/vit/modeling_vit.py @@ -684,7 +684,7 @@ def forward( >>> bool_masked_pos = torch.randint(low=0, high=2, size=(1, num_patches)).bool() >>> 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.reconstructed_pixel_values >>> list(reconstructed_pixel_values.shape) [1, 3, 224, 224] ```""" From b0de38dd4a382948052c5bebf090da4ac513f628 Mon Sep 17 00:00:00 2001 From: ydshieh Date: Wed, 15 Mar 2023 17:25:01 +0100 Subject: [PATCH 2/2] fix --- src/transformers/models/vit/modeling_vit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/models/vit/modeling_vit.py b/src/transformers/models/vit/modeling_vit.py index 3bfdefb08b66..4a7b40409383 100644 --- a/src/transformers/models/vit/modeling_vit.py +++ b/src/transformers/models/vit/modeling_vit.py @@ -684,7 +684,7 @@ def forward( >>> bool_masked_pos = torch.randint(low=0, high=2, size=(1, num_patches)).bool() >>> outputs = model(pixel_values, bool_masked_pos=bool_masked_pos) - >>> loss, reconstructed_pixel_values = outputs.loss, outputs.reconstructed_pixel_values + >>> loss, reconstructed_pixel_values = outputs.loss, outputs.reconstruction >>> list(reconstructed_pixel_values.shape) [1, 3, 224, 224] ```"""