ShoufaChen / AdaptFormer

[NeurIPS 2022] Implementation of "AdaptFormer: Adapting Vision Transformers for Scalable Visual Recognition"

Home Page:https://arxiv.org/abs/2205.13535

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Evaluation

JiuqingDong opened this issue · comments

Hi, I can train by this command:

image

python3 -m torch.distributed.launch --nproc_per_node=2 --use_env main_image.py
--batch_size 64 --cls_token
--finetune ./pretrained/mae_pretrain_vit_b.pth
--dist_eval --data_path ./data/
--output_dir ./output/
--drop_path 0.0 --blr 0.1
--dataset cifar100 --ffn_adapt

After the training process, I got a lot of checkpoints.pth. However, you provide the test command. Could you give some suggestions for test commands?

Hi,Hava you solved this problem?