pytorch 0.3.1 runs well

[xjock]

48G GPU memory，why need so much memory？

update，in pytorch 0.3.1 the code runs well