Dimension out of range

[ChaoyiZh]

Thank you for your great work. But when trying to train the model to reproduce your result, I met a problem.

line 201, in colormap color_map = colors[indices].transpose(2, 3).transpose(1, 2) RuntimeError: Dimension out of range (expected to be in range of [-3, 2], but got 3)

I do not know how to fix it. Looking forward to hear from you! Thank you so much!

[abbaskali]

@ChaoyiZh Dear Chao your problem is almost solved in the below link: https://github.com/pytorch/pytorch/issues/5554