When I run train.py

[hawk90]

pytorch version: 0.4.0 pandas version: 0.24.1

python3 train.py config/example.yaml 123 --> Experiment: example_sd_123 Training Set: Total 2 files Malware Count : 1 Goodware Count: 1 Validation Set: Total 2 files Malware Count : 1 Goodware Count: 1 Traceback (most recent call last): File "train.py", line 151, in history['tr_loss'].append(loss.cpu().data.numpy()[0]) IndexError: too many indices for array

[hawk90]

I changed it.

File "train.py", line 151, in history['tr_loss'].append(loss.cpu().data.numpy()[0]) --> history['tr_loss'].append(loss.cpu().data.numpy())

File "train.py", line 183, in history['val_loss'].append(loss.cpu().data.numpy()[0]) --> history['val_loss'].append(loss.cpu().data.numpy())