JunYeopLee / fast-autoaugment-efficientnet-pytorch

A Pytorch implementation of Fast AutoAugment and EfficientNet
119 stars 13 forks source link

AttributeError: 'Args' object has no attribute 'use_seblock' #9

Open jeffreygaoo opened 3 years ago

jeffreygaoo commented 3 years ago

HI,Thanks a lot for your code.And i met this error as the titile shown(AttributeError: 'Args' object has no attribute 'use_seblock') when i trained the network. [+] Create network Traceback (most recent call last): File "/home/leinao/models/efficientnet/train.py", line 98, in fire.Fire(train) File "/home/leinao/.local/lib/python3.9/site-packages/fire/core.py", line 141, in Fire component_trace = _Fire(component, args, parsed_flag_args, context, name) File "/home/leinao/.local/lib/python3.9/site-packages/fire/core.py", line 466, in _Fire component, remaining_args = _CallAndUpdateTrace( File "/home/leinao/.local/lib/python3.9/site-packages/fire/core.py", line 681, in _CallAndUpdateTrace component = fn(*varargs, **kwargs) File "/home/leinao/models/efficientnet/train.py", line 35, in train model = select_model(args) File "/home/leinao/models/efficientnet/utils.py", line 112, in select_model model = Net(args) File "/home/leinao/models/efficientnet/networks/efficientnet_cifar10.py", line 156, in init self.use_seblock = args.use_seblock AttributeError: 'Args' object has no attribute 'use_seblock'