seems don't run on GPU??

sunchang0124 / dp_cgans

A library to generate synthetic tabular or RDF data using Conditional Generative Adversary Networks (GANs) combined with Differential Privacy techniques.

MIT License

34 stars 10 forks source link

seems don't run on GPU?? #6

Open caprone opened 1 year ago

caprone commented 1 year ago

HI i'm try to train a base model, but seems does't works with GPU?..is very slow and no output from (verbose =True)... any idea? Thanks

sunchang0124 commented 12 months ago

Hi Sorry for the late reply. It does run on the GPU when GPU is available in your system. Did you check if you have CUDA? I am going to update a version here you can monitor the training process so you can see if the GPU is being used or not. Thank you.