search

prajwalsingh / EEGStyleGAN-ADA

Pytorch code of paper "Learning Robust Deep Visual Representations from EEG Brain Recordings". [WACV 2024]
MIT License
27 stars 4 forks source link

About hardware #10

Closed 7ASSEL closed 6 months ago

7ASSEL commented 6 months ago

Thanks for your excellent work! Could you please tell me what GPU and VRAM do we need to train the model?

prajwalsingh commented 6 months ago

Hi @7ASSEL , thank you for reading our work.

We used Rtx Quadro 5000 (16GB) for training and inference of EEG features, and for training StyleGAN-ADA, we used Tesla V100 (32GB).

7ASSEL commented 6 months ago

Hi @7ASSEL , thank you for reading our work.

We used Rtx Quadro 5000 (16GB) for training and inference of EEG features, and for training StyleGAN-ADA, we used Tesla V100 (32GB).

Thanks a lot!