如果要运行pretrain:
python pretrain.py --max_epoch 200 --episodes_per_epoch 1000 --model_class PreMod --use_euclidean --backbone_class Res18 --D 512 --dataset cifar100 --num_classes 100 --way 5 --shot 1 --eval_shot 1 --query 15 --eval_query 15 --init_lr 0.00001 --lr_scheduler step --init_weights xxx --eval_interval 1 --step_size 20 --gamma 0.5 --beta 0.01 --batch_size 10 --save_dir yyy
其中--init_weight代表加载预训练模型文件的位置,--save_dir表示存放结果的根目录
如果要运行metatrain:
python metatrain.py --dataset cifar10 --num_workers 0 --log_interval 2 --episodes_per_epoch 1000 --model_class MetaMod --backbone_class Res12 --init_weights D:\_doors_programs\fsl-save\cifar10-PreMod-Res12-05w01s15q-Pre-Dis\20_0.5_lr1e-05_step_T10.1_b0.01_ba_sz160-NoAug\max_acc.pth --use_euclidean --batch_size 10 --init_lr 0.0001 --lr_scheduler step --step_size 40 --query 5 --eval_query 5