I train nn on jupyter notebook,
at the beginning it takes 1 min per epoch,
but the time per epoch increase little by little.
after 10 hours will become 2min per epoch.
That is why?
I search on google, found that many person have this problem.
And many person thought that is jupyter notebook’s problem.
How to do with it??
What nvidia Driver are you using ? Your question is too generic !
Its difficult to help you out holding information with you. I don’t see any problem with WGAN at all that could slow down your training process.
May be is a bad implementation of the network , or your data is bad distributed or even your learning rate is not good enough.Also my question were important and you may think not but in fact the Nvidia driver 390 had memory leak slowing down applications overtime.
Without any clue from your side is very difficult to debate anything to you because I don’t know nothing about your model and nothing about your implementation and data…
Man I am cool … my question is … do you really want some help ?
Assuming that your driver is bigger than version 390 I am sure its not related your driver.
You probably using conda right ? Well you can try to upgrade cuda version to 9.2 and cudnn because cudnn is the responsible to accelerate the cuda itself. And update your Pytorch to version 1.0 or 0.41
If that don’t work then the lastest thing to blame will be the code itself, but I just wanted to reduce the probability of problems related to drivers first.