boschresearch / Continuous-Recurrent-Units

GNU Affero General Public License v3.0
53 stars 10 forks source link

Running Code on Windows 11 #5

Open dubeypoornima20 opened 1 year ago

dubeypoornima20 commented 1 year ago

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?

monasch commented 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.