Hello all,
I want to keep around the best model while training without writing it to disk every time a new best model is encountered, which is what I’m currently doing.
Can I copy the model that is on the GPU over to the CPU somehow?  I don’t want to make an extra copy on the GPU since I need that memory.
Thanks!