torch.save(net.state_dict(), 'cifar_net.pth')保存的文件的绝对路径
时间: 2024-03-06 08:50:33 浏览: 47
遍历并输出指定文件的绝对路径
如果您在运行这行代码时未指定保存文件的绝对路径,则文件将保存在当前工作目录中。要获取当前工作目录,可以使用Python中的os模块。以下是一个示例代码,可以获取当前工作目录并打印出保存的文件的绝对路径:
```
import os
# 获取当前工作目录
cwd = os.getcwd()
# 拼接文件路径
file_path = os.path.join(cwd, 'cifar_net.pth')
print("文件保存在:", file_path)
```
这段代码将打印出保存文件的绝对路径,例如:`文件保存在:/home/user/cifar_net.pth`。
阅读全文