加载/保存 .pt 文件

.pt类型的扩展文件常用来保存张量;

加载文件

import torch

# Accuracy
# Accuracy = torch.load('path')
Accuracy = torch.load('experiments/Accuracy.pt')
print("Accuracy:")
print(Accuracy)

在这里插入图片描述

保存

import torch

# path写在右边
torch.save(pt,'Accuracy.pt')

猜你喜欢

转载自blog.csdn.net/qq_43585922/article/details/129460955
pt