Hi, I am trying to train a relatively large model on GPU, and do some memory-intensive computation at the end of each epoch which cannot be done on GPU.  Currently I am converting model to cpu at the end of each epoch, do the computation and then convert it back to gpu.  But it would take relatively long time for the conversion and computation, and there is really no need to wait for the computation to finish before training next epoch.  So I am wondering whether it is possible to copy the model to cpu and let it run the computation asynchronously, without interrupting the training on GPU?
Thank you!