使用PyTorch實現resnet-18分類CIFAR-10

昨天看完了resnet的論文,今天試着來實現一下。web 先放一個resnet18的模型圖: 網絡 模塊引入 # 模塊引入 import torch import torchvision import torchvision.transforms as transforms import torch.nn as nn import torch.nn.functional as F import
相關文章
相關標籤/搜索