利用nn.ModuleList()代替List
Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same
最新推荐文章于 2024-06-06 18:57:42 发布
利用nn.ModuleList()代替List