
pyTorch学习
资瓷向量机
Alibaba
展开
-
pyTorch学习2:搭建网络
利用pytorch搭建网络的核心在于继承类torch.nn.Module并重写__init__()和forward()函数,添加新层的方式有如下几种:直接添加self.conv1 = torch.nn.Conv2d(...)self.fc1 = torch.nn.Linear(...)add_module方法self.add_module("conv1", torch.nn.C...原创 2019-11-25 17:52:35 · 192 阅读 · 0 评论 -
pyTorch学习1:环境配置
1、添加国内镜像参考:conda安装Pytorch下载过慢解决办法conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/m...原创 2019-11-22 16:48:16 · 147 阅读 · 0 评论