 run train demo : can't convert cuda:0 device type tensor to numpy.
