PyTorch學習之路:ResNet-34實現CIFAR10分類

本代碼參考廖星宇《深度學習入門之PyTorch》中的示例代碼,手動拼接復現而來,僅供個人使用,侵刪。 #ResNet實現CIFAR10分類 from datetime import datetime import torch import torch.nn.functional as F from torch import nn from torch.autograd import Variabl
相關文章
相關標籤/搜索