![](https://img-blog.csdnimg.cn/20201014180756913.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
pytorch
Datura_Metel
这个作者很懒,什么都没留下…
展开
-
了解网络模型的参数
1、打印网络每层输出的大小import torchimport torch.nn as nnfrom torchsummary import summaryclass Net(nn.Module): def __init__(self): super(Net, self).__init__() self.fc1 = nn.Linear(10, 5) self.fc2 = nn.Linear(5, 10) def forward原创 2020-12-14 14:00:34 · 680 阅读 · 1 评论 -
Windows 预训练BERT模型的安装、加载
安装1、通过Anaconda创建虚拟环境conda create -n python python=3.82、查看所有的虚拟环境conda env list3、激活虚拟环境python38activate python384、(pytorch需要在此步骤之前安装,我的是1.5.1版本的pytorch)pip安装预训练BERT模型pip install pytorch-pretrained-bert5、下载预训练模型和相应的词表6个不同的模型(根据需要原创 2020-12-01 14:10:46 · 1702 阅读 · 0 评论