IndexError: too many indices for array: array is 0-dimensional, but 1 were indexed

[yzj0531]

File "train.py", line 355, in main
loss_semi_adv_value += loss_semi_adv.data.cpu().numpy()[0]/args.lambda_semi_adv
IndexError: too many indices for array: array is 0-dimensional, but 1 were indexed

i need help, thanks!

[yeeyangtee]

you can try:
loss_semi_adv_value += loss_semi_adv.data.cpu().numpy().item()/args.lambda_semi_adv

For newer version of PyTorch, this is better:
loss_semi_adv_value += loss_semi_adv.item()/args.lambda_semi_adv