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'
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'