使用pytorch版本跑NYUD数据集时需要更改那些地方的代码呢

[lucy11111111]

我用pytorch版本直接跑NYUD数据集时会有报错，具体为axes don't match array和Caught ValueError in DataLoader worker process 0.
想请问一下跑NYUD时代码时需要更改那些地方吗？
另外平均RGB模型和HHA模型的输出具体是怎么做呢？
合并HHA、RGB的意思是将HHA和RGB的图片放到一起训练吗？