too many indices for array

[yueanga]

when i ran with

python main.py --train_folder </.../...> --test_folder</.../...>

an error occured in around line 161

loss_nle_mean(torch.mean(nle_value).data.cpu( ).numpy( )[ ])
IndexError:too many indices for array: array is 0-dimensional,but 1 were indexed

how can i solve this? thank you!

[liaini]

loss_nle_mean(torch.mean(nle_value).data.cpu().numpy()) works~

[zideliu]

have you run this code succeedly?
I meet "inplace operation" error,