Dimbl4

You're computing on the cpu. Try to add "--device cuda", if you have one.

@edebrouwer try to upgrade torch 1.4.0 to nightly, it helped me. (cuda 10.0, torch 1.5.0.dev20200313+cu100)