1.RuntimeError: Found 0 images in subfolders of: ./data/aaa
原因分析: 文件夹aaa下没有子文件夹报错
解决问题: 在文件夹aaa建立符合要求的自文件夹,符合数据格式。
(类似这种)
2.UserWarning: Detected call of lr_scheduler.step()
before optimizer.step()
. In PyTorch 1.1.0 and later, you should call them in the opposite order: optimizer.step()
before `lr_scheduler.step()’**
原因分析:由于PyTorch(从PyTorch> = 1.1.0开始)的版本升级,对sdcheduler.step()和optimizer.step()顺序有要求。
解决问题:如警告所述,调用sdcheduler.step()应该移动到调用optimizer.step()之后即可。
3.RuntimeError: CUDA error: device-side assert triggered – 1D CVAE
原因分析:不清楚
解决问题:使用以下命令运行该模型:
import os
os.environ[‘CUDA_LAUNCH_BLOCKING’] = “1”
- 可以使用下面这行代码来简单检查网络结构。
print(model)
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。
文章由极客之音整理,本文链接:https://www.bmabk.com/index.php/post/99878.html