Pytroch學習筆記(3)優化器|莫凡python

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), dim
相關文章
相關標籤/搜索