Vivek Chandra

Downgrading torch and torchvision to 0.2.0 and 0.2.1 solved this issue for me.

@TGISer, @wgqtmac, @Tianlin-Gao - seems like you all got it to work. There's no change to `main.py` since you got it to work. Any pointers on changes you made for...

Seems like same as #20 but with more details. Also, i'm facing the same error.

@Dr-Zhou Do you mean, solution can be provided on Stackoverflow? OR the solution is already there? -- there are no answers for the Question.

Downgrading torch and torchvision to 0.2.0 and 0.2.1 solved this issue for me.

Were u able to solve this?