pytorch代码学习
关注
分享
扫一扫
BigCowPeking
追求卓越，永不放弃
展开
-
pytorch 自定义网络结构 +自定义数据加载
自定义网络结构 import osos.environ['CUDA_VISIBLE_DEVICES'] = '3'os.system('echo $CUDA_VISIBLE_DEVICES') import torchimport torch.nn as nnimport torch.nn.functional as Fimport torch.utils.data as dat...转载 2019-08-18 18:19:56 · 643 阅读 · 0 评论 -
pytorch学习-数据处理生成batch
开始学习PyTorch，在此记录学习过程。准备按顺序写以下系列：加载数据并生成batch数据数据预处理构建神经网络Tenso...转载 2019-08-18 18:03:29 · 6132 阅读 · 0 评论