卷积神经网络(demo)
摘要:
import torch import torchvision as tv import torchvision.transforms as transforms import torch.nn as nn import torch.optim as optim import argparse # 定义是否使用GPU device = torch.device("cuda" if torch.c... 阅读全文
posted @ 2019-07-23 15:17 茯苓 阅读(402) 评论(0) 推荐(0) 编辑