search

changlin31 / DNA

(CVPR 2020) Block-wisely Supervised Neural Architecture Search with Knowledge Distillation
235 stars 35 forks source link

您好，我在运行searching/train.py时出现了一个错误 #14

Closed doer-hjh closed 4 years ago

doer-hjh commented 4 years ago

错误显示如下： File "D:/test/DNA-master-new/DNA-master/searching/train.py", line 277, in main() File "D:/test/DNA-master-new/DNA-master/searching/train.py", line 272, in main writer=writer) File "D:\test\DNA-master-new\DNA-master\searching\dna\distill_train.py", line 104, in distill_train reset_data=reset_data) File "D:\test\DNA-master-new\DNA-master\searching\dna\distill_train.py", line 699, in _potential for layer in supernet.module.modules(): File "D:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 594, in getattr type(self).name, name)) AttributeError: 'StudentSuperNet' object has no attribute 'module' 还请您能够帮忙解答！

changlin31 commented 4 years ago

We only support training with Pytorch Distributed Data Parallel (DDP) mode. Run your code with sh dist_train.sh even if you are using single GPU (remember to set nproc_per_node in dist_train.sh to 1).

doer-hjh commented 4 years ago

Thank you very much for your reply