System Requirements

Open robjlyons opened this issue 3 years ago • 1 comments

Hi, what are the systems requirements for this?

I'm using a 3070Ti and am running out of cuda memory.

Thanks,

Sep 05 '22 09:09 robjlyons

Hey, the default configuration should require around 34 GB of memory. You can reduce the memory requirements by decreasing the batch size batch_num_samples and increasing the number of gradient accumulation steps grad_acc_steps in config/trainer.yaml.

Sep 05 '22 12:09 vmicheli