摘要: 试一试 1 import torch 2 import torch.nn as nn 3 import torch.nn.functional as F 4 from torchsummary import summary 5 6 class ResBlock(nn.Module): 7 def _ 阅读全文
posted @ 2021-04-02 16:43 一剑光寒十四州 阅读(760) 评论(0) 推荐(0) 编辑
摘要: 1 import os 2 import torch 3 from torchvision import transforms 4 from data_pipe import get_data 5 from vgg import VGG_13 6 from resnet18 import ResNe 阅读全文
posted @ 2021-04-02 16:40 一剑光寒十四州 阅读(87) 评论(0) 推荐(0) 编辑