使用torch.save时出现RuntimeError: Parent directory C:\Users\86187\Desktop\大学事务\多元数据融合与智能分析\实验二\Amazing-Brick-DFS-and-DRL\DQN_train\..\DQN_train\dqn_weights does not exist.,但是我电脑中这个目录一直存在
时间: 2023-07-01 15:28:15 浏览: 1472
解决pycharm导入numpy包的和使用时报错:RuntimeError: The current Numpy installation (‘D:\\python3.6\\lib\\site-packa的问题
这个错误可能是由于你指定的文件夹路径中包含了".."这种相对路径导致的。如果你使用相对路径作为文件夹路径,可能会遇到这种问题,因为程序无法确定相对路径的确切位置。
为了避免这个问题,建议使用绝对路径作为文件夹路径。可以使用 `os.path.abspath()` 函数将相对路径转换为绝对路径,例如:
```
import os
import torch
model = MyModel()
folder_path = os.path.abspath('dqn_weights')
if not os.path.exists(folder_path):
os.makedirs(folder_path)
torch.save(model.state_dict(), os.path.join(folder_path, 'model.pth'))
```
这里使用了 `os.path.abspath()` 将相对路径转换为绝对路径,然后使用 `os.path.join()` 函数将文件名和文件夹路径拼接在一起。同时,使用了 `os.makedirs()` 函数创建了文件夹,如果文件夹不存在的话。
阅读全文