newsreclib icon indicating copy to clipboard operation
newsreclib copied to clipboard

Can I run a job with pytorch distributed training?

Open chiyuzhang94 opened this issue 8 months ago • 1 comments

Can I run a job with pytorch distributed training? If I run this commend, does it work? torchrun --nproc_per_node=$WORLD_SIZE --master_port=1234 newsreclib/train.py experiment=nrms_mindsmall_pretrainedemb_celoss_bertsent

chiyuzhang94 avatar Dec 19 '23 18:12 chiyuzhang94