Jolen_xie的博客

import torchimport torchvision.models as models #储存里常用的一些模型import torch.nn as nnimport torch.optim as optimimport torch.optim.lr_scheduler as lr_schedulerimport timeimport torchvision.transforms as transforms # 常用的图片变换，例如裁剪、旋转等；from torchvision impo

import torchfrom torch.autograd import Variableimport numpy as npimport timeuse_cuda = Truedef get_data(): # 生成拟合数据 train_x = np.asarray([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20]) #[10] train_y = np.asarray([1, 2