From 18fc2808e04bdfa51e845f4771c5704e937a4f13 Mon Sep 17 00:00:00 2001 From: Nathan Somavarapu Date: Mon, 1 Mar 2021 23:30:47 -0500 Subject: [PATCH] Update model.py --- pytorch_pretrained_vit/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_pretrained_vit/model.py b/pytorch_pretrained_vit/model.py index 22cfef9..aef7373 100755 --- a/pytorch_pretrained_vit/model.py +++ b/pytorch_pretrained_vit/model.py @@ -61,7 +61,7 @@ def __init__( if name is None: check_msg = 'must specify name of pretrained model' assert not pretrained, check_msg - assert not resize_positional_embedding, check_msg +# assert not resize_positional_embedding, check_msg if num_classes is None: num_classes = 1000 if image_size is None: