Cannot run

[razvanbugariu]

Hi,
@lizeng614
I am trying to run it on Google Colab, but after the message from first iteration is displayed, I am receiving a Killed message. Any thoughts?

[weizhenhuan]

Hi, I am also trying to run it on Colab, but I get
IndexError: invalid index of a 0-dim tensor. Use tensor.item() in Python or tensor.item<T>() in C++ to convert a 0-dim tensor to a number
Any thoughts?
@lizeng614