Hello, I use GPU (cuda 11.1) to train a massive IA model based on pytorch, but while training the time of every epoch is contantly increasing (starting by 20 s on going up) I do not why, can you help please?
here is the training loop.
thanks