@LOVELWY
文章目录
- 前言
- 一、TypeError: load() missing 1 required positional argument: ‘Loader‘?
- 二、解决pytorch报错:AssertionError: Invalid device id
- 三、 if __name__ == '__main__':报错
- 四、RuntimeError: CUDA out of memory.
- 五、IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item<T>()` in C++ to convert a 0-dim tensor to a number
- 总结:运行成功
前言
提示:Windows环境、Geforce RTX 3060Ti、Pytorch1.7.0:
记录EnlightenGAN过程中的问题和解决方式。
提示:以下是本篇文章正文内容,下面案例可供参考
一、TypeError: load() missing 1 required positional argument: ‘Loader‘?
出错原因:请求API的时候缺少参数
1.降低yaml版本
Yaml 5.1版本之后就弃用了不安全的load加载方式,需要在load方法中加入指定的加载器,或者使用安全加载api, 通过默认加载器使load函数的安全得到加强。
安装低版本的yaml库即可
pip install pyyaml==5.1
2.修改代码
代码如下(任选三条中的一条修改):
yaml.