pytorch 在我的ADAM算法中生成此维度错误在哪里？ 返回

在我的ADAM算法中生成此维度错误在哪里？


生如夏花
2025-03-15 01:28:43 (7天前)


我正在使用这个BRATS Unet中的main.py

https://github.com/pykao/Modified-3D-UNet-Pytorch/blob/master/main.py

＃创建优化器
打印（“创建优化器”）

optimizer = optim.adam（net ….


2 条回复
  1. 0# 冷月如霜·胡狼 | 2019-08-31 10-32



    您应该将数据集传递给数据加载器API。因此，将train_data和test_data而不是train_idx和test_idx传递给torch.utils.data.DataLoader。


登录 后才能参与评论