Open helloworld-wu opened 2 years ago
When I want to use the pretrained model, I got a problem likes this. Could you tell me how to slove it? Thanks!
把这行改成https://github.com/yitu-opensource/T2T-ViT/blob/main/utils.py#L83
state_dict = load_state_dict(checkpoint_path, model, use_ema, num_classes)
就行了 @helloworld-wu
把这行改成https://github.com/yitu-opensource/T2T-ViT/blob/main/utils.py#L83
state_dict = load_state_dict(checkpoint_path, model, use_ema, num_classes)
就行了 @helloworld-wu
谢谢!