@moskomule Hi I got this error after run it.
How to fixed this?
You want to upgrade to pytorch 0.2 (for which the used Variable.grad_fn replaces the older and obsolete Variable.creator).
Does this work as of pytorch v0.2? I tried to run with a simple sequential model and got an empty image
The function wants a Variable, e.g. a module calculation result. That worked for me with 0.2.