每日查询知识归笼

2020年12月27日
图像处理库Skimage安装与使用

torch加载与保存模型：

# Specify a path
PATH = "state_dict_model.pt"

# Save
torch.save(net.state_dict(), PATH)

# Load
model = Net()
model.load_state_dict(torch.load(PATH)) #先进行序列化
model.eval()

#保存整个模型
# Specify a path
PATH = "entire_model.pt"

# Save
torch.save(net, PATH)

# Load
model = torch.load(PATH)
model.eval()

numpy中更改维度：

array.transpose()

保存array到本地：

np.save('demo.npy',m) #名字，变量
np.load("demo.npy")

2020年12月31日

大电脑上，激活环境使用 conda  activate base

torch中make_grid()后进行画图时，要转化为array，以及通道，cpu

搞懂iterable  iter() __getitem()__ next()等