2021
07-31
07-31
浅谈Pytorch 定义的网络结构层能否重复使用
前言:最近在构建网络的时候,有一些层参数一样,于是就没有定义新的层,直接重复使用了原来已经有的层,发现效果和模型大小都没有什么变化,心中产生了疑问:定义的网络结构层能否重复使用?因此接下来利用了一个小模型网络实验了一下。一、网络结构一:(连续使用相同的层)1、网络结构如下所示:classCnn(nn.Module):def__init__(self):super(Cnn,self).__init__()self.conv1=nn.Sequential(...
继续阅读 >