Skip to content

vandoor/fewshot-CVET

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

如果要运行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

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages