Giters

yitu-opensource / T2T-ViT

ICCV2021, Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Bug in utils.py ->load_for_transfer_learning

AmnonGeifman opened this issue · comments

commented

When using load_state_dict utile function the model needs to be given as input to the function. The first example on the readme doesn't work otherwise.