icoz69 / DeepEMD

Code for paper "DeepEMD: Few-Shot Image Classification with Differentiable Earth Mover's Distance and Structured Classifiers", CVPR2020
MIT License
577 stars 82 forks source link

Missing some sfc arguments? #54

Open sitns opened 3 weeks ago

sitns commented 3 weeks ago

when trying to train a 5 way 5 shot model in train_pretrain.py,I found the following error occured: File "/root/DeepEMD/Models/models/Network.py", line 89, in get_sfc for k in range(0, self.args.sfc_update_step): AttributeError: 'Namespace' object has no attribute 'sfc_update_step' i add argument 'sfc_lr' and the following 'sfc_bs'\ 'sfc_update_size' ,then it started fuctioning properly. Is that just a hiccup or there's something i miss?