最近使用pytorch训练网络,发现一个有意思的事情。
意思的事情
我希望保存效果最好的模型参数,但是保存路径上包含中文(Chinese characters),如果使用 torch.save() 函数,则出现如下错误。
例如:保存路径为
saved_model_dir = 'E:/新建文件夹/saved_model/xxx.pt'
则在如下地方
torch.save(saved_model_dict, saved_model_dir)
出现错误
如果修改为
saved_model_dir = 'E:/new_folder/saved_model/xxx.pt'
则可以顺利保存!!
总结:
使用torch.save() 函数时保存路径不能含有中文。
分析:
之所以没有这个觉悟,是因为之前在读取或者保存csv/txt文件的时候,采用 pandas 库
import pandas as pd
pd.read_csv()
pd.to_csv()
亦或者使用 csv 模块
import csv
with open('filename.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
# 处理每一行数据
print(row)
的时候也都是中英混着用,从未关注这个问题。
看来torch还需要再改进一步 :)