05 2021 档案
module
摘要:import torchfrom torch import nnfrom torch import optim#优化器类# 1. torch.optim.SGD(参数,学习率)# 2. torch.optim.Adam(参数,学习率)import numpy as npfrom matplotlib 阅读全文
posted @ 2021-05-10 22:39 Sleven 阅读(160) 评论(0) 推荐(0)
train_data
摘要:for images, labels in train_data: for images, labels in train_data: img = images[0] img = img.numpy() img = np.transpose(img, (1, 2, 0)) plt.imshow(im 阅读全文
posted @ 2021-05-03 15:24 Sleven 阅读(119) 评论(0) 推荐(0)