invalid index of a 0-dim tensor

[Jakexxh]

When I run the model with CUP, and got this error:

Traceback (most recent call last):
File "train.py", line 38, in
errors = model.get_current_errors()
File "/Users/XXX/Desktop/XXX/MC-GAN/models/cGAN_model.py", line 250, in get_current_errors
return OrderedDict([('G_GAN', self.loss_G_GAN.data[0]),
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number`

Please change the code return OrderedDict([('G_GAN', self.loss_G_GAN.data[0]) in line 250 of cGAN_model.py into return OrderedDict([('G_GAN', self.loss_G_GAN.data)

[narusevic]

@Jakexxh Have you solved this issue?

[narusevic]

NVIDIA/flownet2-pytorch#113 (comment) @Jakexxh

[Jakexxh]

Yes