Traceback (most recent call last):
File "g:\GitHub_work\YOLOv4-pytorch-jingtianyilong\train.py", line 289, in
Trainer(log_dir,resume=args.resume, fine_tune=args.fine_tune).train()
File "g:\GitHub_work\YOLOv4-pytorch-jingtianyilong\train.py", line 143, in train
if self.fp_16: self.yolov4, self.optimizer = amp.initialize(self.yolov4, self.optimizer, opt_level='O1', verbosity=0)
File "H:\Anaconda3\lib\site-packages\apex\amp\frontend.py", line 357, in initialize
return _initialize(models, optimizers, _amp_state.opt_properties, num_losses, cast_model_outputs)
File "H:\Anaconda3\lib\site-packages\apex\amp_initialize.py", line 241, in _initialize
handle = amp_init(loss_scale=properties.loss_scale, verbose=(_amp_state.verbosity == 2))
File "H:\Anaconda3\lib\site-packages\apex\amp\amp.py", line 101, in init
try_caching, verbose)
File "H:\Anaconda3\lib\site-packages\apex\amp\wrap.py", line 33, in cached_cast
if not utils.has_func(mod, fn):
File "H:\Anaconda3\lib\site-packages\apex\amp\utils.py", line 132, in has_func
if isinstance(mod, torch.nn.backends.backend.FunctionBackend):
AttributeError: module 'torch.nn' has no attribute 'backends
Traceback (most recent call last): File "g:\GitHub_work\YOLOv4-pytorch-jingtianyilong\train.py", line 289, in
Trainer(log_dir,resume=args.resume, fine_tune=args.fine_tune).train()
File "g:\GitHub_work\YOLOv4-pytorch-jingtianyilong\train.py", line 143, in train
if self.fp_16: self.yolov4, self.optimizer = amp.initialize(self.yolov4, self.optimizer, opt_level='O1', verbosity=0)
File "H:\Anaconda3\lib\site-packages\apex\amp\frontend.py", line 357, in initialize
return _initialize(models, optimizers, _amp_state.opt_properties, num_losses, cast_model_outputs)
File "H:\Anaconda3\lib\site-packages\apex\amp_initialize.py", line 241, in _initialize
handle = amp_init(loss_scale=properties.loss_scale, verbose=(_amp_state.verbosity == 2))
File "H:\Anaconda3\lib\site-packages\apex\amp\amp.py", line 101, in init
try_caching, verbose)
File "H:\Anaconda3\lib\site-packages\apex\amp\wrap.py", line 33, in cached_cast
if not utils.has_func(mod, fn):
File "H:\Anaconda3\lib\site-packages\apex\amp\utils.py", line 132, in has_func
if isinstance(mod, torch.nn.backends.backend.FunctionBackend):
AttributeError: module 'torch.nn' has no attribute 'backends
作者你好,这个错误怎么解决呢,没找到办法