莫煩pytorch(9)——optimizer

1.畫圖 import torch import torch.utils.data as Data import torch.nn.functional as F import matplotlib.pyplot as plt LR = 0.01 BATCH_SIZE = 32 EPOCH = 12 x=torch.unsqueeze(torch.linspace(-1,1,1000),di
相關文章
相關標籤/搜索