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: