How to solve pytorch freezing issue?

[ptrblck]

This tutorial might be a good starter.