worldbank/REaLTabFormer

Multi-GPU training

vinay-k12 opened this issue · 1 comments

I was trying to run training on multiple GPU servers in AWS, but it is not training as expected. Is there a way to enable this?

Hi there!

Yes, it seems now that for some reason it is not working while n the past training was utilizing multi-GPU.

Can anyone verify this as well?

@avsolatorio, maybe it is because FrozenSeq2SeqTrainer was replaced with Seq2SeqTrainer?

Thanks!