Pytorch: parameters(),children(),modules(),named_*區別

nn.Module vs nn.functional 前者會保存權重等信息,後者只是作運算html parameters() 返回可訓練參數python nn.ModuleList vs. nn.ParameterList vs. nn.Sequential layer_list = [nn.Conv2d(5,5,3), nn.BatchNorm2d(5), nn.Linear(5,2)] cl
相關文章
相關標籤/搜索