在pytorch中可能层数不确定,需要用一个列表的形式进行模型构建。如果直接在Module里面写self.xx=[层1,层2]
这样就不能将这些参数放到模型可训练参数列表中。需要用self.xx=torch.nn.ModuleList([层1,层2])
pytorch模型列表的处理 2021-09-06
最新推荐文章于 2023-06-25 17:35:10 发布
在pytorch中可能层数不确定,需要用一个列表的形式进行模型构建。如果直接在Module里面写self.xx=[层1,层2]
这样就不能将这些参数放到模型可训练参数列表中。需要用self.xx=torch.nn.ModuleList([层1,层2])