AdvSemiSeg icon indicating copy to clipboard operation
AdvSemiSeg copied to clipboard

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

Open yzj0531 opened this issue 4 years ago • 1 comments

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!

yzj0531 avatar Nov 19 '20 09:11 yzj0531

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

yeeyangtee avatar Jan 18 '21 07:01 yeeyangtee