search

jingtianyilong / YOLOv4-pytorch

This is a pytorch repository of YOLOv4 that can be trained with custom dataset.
31 stars 12 forks source link

AttributeError: module 'torch.nn' has no attribute 'backends #6

Closed leiqing110 closed 3 years ago

leiqing110 commented 3 years ago

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

作者你好，这个错误怎么解决呢，没找到办法