too many indices for tensor of dimension 1

  File "vis_joint_inference.py", line 78, in <module>
    bbox_selected, label_orig, opt_maskgen)
  File "/home/ubuntu/user_space/neurips18_hierchical_image_manipulation-master/models/joint_inference_model.py", line 75, in gen_layout
    }, target_size=(input_dict['label_orig'].size()[2:4]))
  File "/home/ubuntu/user_space/neurips18_hierchical_image_manipulation-master/models/TwoStreamAE_mask.py", line 323, in evaluate
    self.encode_input(label_map, mask_ctx_in, mask_out, mask_in, cls)
  File "/home/ubuntu/user_space/neurips18_hierchical_image_manipulation-master/models/TwoStreamAE_mask.py", line 157, in encode_input
    oneHot_obj_mask_in[b,cls.data[b,0],:,:] = mask_in.data[b,0]
IndexError: too many indices for tensor of dimension 1

indices类型错误吧，转成long试试