pytorch Fix n fold cross validation code add some measure and plots

I am looking for a freelancer to Fix n-fold cross validation to my Convolutional Neural Network (CNN) deep learning models.
using pytorch
I have some notebooks containing simple error code
also I want to fix a comparison file (ROC, PR, ...) for the output pickles files that output from these notebooks.
the required is to fix n fold cross validation to these notebooks.
then fix the figures (ROC, PR,..) and output values to show the effect of n fold cross validation.