Efficient Backprop for Virtual Adversarial Training

[tom (Thomas V)]

Would passing retain_graph not work?