nupurkmr9 / S2M2_fewshot

Other
112 stars 20 forks source link

How to pre-train the model #30

Open zhang1hongliang opened 1 year ago

zhang1hongliang commented 1 year ago

@nupurkmr9, When I pre-train the WideResNet28_10 model for S2M2 on cifar-fs, I come into some problems, such as it needs to load resume_file by torch.load function in line 273, but there is no .tar file in the pre-training process, in addition, I need you to tell me how to pre-train the model on cifar-fs and its corresponding parameter settings such as batch size and epochs.