Open mrxgavin opened 1 year ago
Hi, Thanks for your interest.
I can manage to run the model using an Nvidia V100 with 32 GB configuration.
I don't use batch training in this model. If you want to run the model with less CUDA memory, you can try to modify hyper-parameters associated with the graph transformer model, such as "head" and "embed_size".
Hi, may I ask how much GPU memory is needed to train the model? Is there any way to modify the batch size?