Open dubeypoornima20 opened 1 year ago
Hi, 12-13 minutes per epoch sounds to me like your training on CPU. You might want to use a device with GPUs to accelerate the training. The training times from the paper are timed on Nivida Quatro RTX 6000/8000.
Hello,
I am attempting to reproduce the results from your GitHub code on my Windows 11 machine, but I am experiencing slow training times. Specifically, each epoch is taking around 12-13 minutes to complete.
I have checked my hardware setup and verified that it meets the minimum requirements specified in the paper/[GitHub code]. How to improve the running speed of the code. Do you have any recommendations for optimizing the code further?