Sample_of_training_process_with_google_colab.ipynb got error

[raychangCode]

Dear author,

I'm using your google colab
I followed your instruction from 'How to use' but I got "IndexError: too many indices for array: array is 1-dimensional, but 2 were indexed" while running train.py.
May I ask you how to solve this?

Thanks