Traceback (most recent call last): File "H:\Swin-Transformer-Object-Detection\tools\train.py", line 179, in <module> main() File "H:\Swin-Transformer-Object-Detection\tools\train.py", line 168, in main train_detector( File "H:\Swin-Transformer-Object-Detection\mmdet\apis\train.py", line 84, in train_detector model, optimizer = apex.amp.initialize( AttributeError: module 'apex' has no attribute 'amp'
时间: 2024-04-03 12:31:32 浏览: 56
rich-traceback:Python 日志记录模块的信息回溯
这个错误是因为你的代码调用了apex库中不存在的amp属性。amp是一个混合精度训练的工具,需要安装apex库并且版本较高才能使用。你需要检查代码中apex库的版本是否正确,并且确认你已经正确安装了apex库。如果你的版本过低,可以尝试更新apex库。另外,也可以尝试使用其他混合精度训练的工具,比如torch.cuda.amp。
阅读全文