-
深度学习的优化器选择(SGD、Momentum、RMSprop、Adam四种)
摘要:import torch import torch.utils.data as Data import torch.nn.functional as F import matplotlib.pyplot as plt import torch.nn as nn LR=0.01 BATCH_SIZE=
阅读全文
-
第一个深度学习网络(别人的)
摘要:import numpy as np import torch from torchvision.datasets import mnist import torchvision.transforms as transforms from torch.utils.data import DataLo
阅读全文
|