Hi Dear @DJ_1992,
There is a similar answer to your inquiry, Hope it helps. 
https://discuss.pytorch.org/t/what-is-the-best-way-to-apply-k-fold-cross-validation-in-cnn/15035/2?u=arminarj
Hi Dear @DJ_1992,
There is a similar answer to your inquiry, Hope it helps.
https://discuss.pytorch.org/t/what-is-the-best-way-to-apply-k-fold-cross-validation-in-cnn/15035/2?u=arminarj