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)