Hi Richard,
Thanks for helping! The dtype is:
dtype = torch.cuda.FloatTensor.
I was thinking about upgrading to 0.3 and see if that solves the problem. But the conda default upgrading seems will replace 0.2 with 0.18, which was a little weird. Do you know a good way to upgrade from 0.2 to 0.3?
Thanks again!