InternImage icon indicating copy to clipboard operation
InternImage copied to clipboard

Unable to train the algorithms with 2 GPU instances (multi-node), each with 4 A100s

Open CheruscanArminius opened this issue 2 years ago • 0 comments

The paper says the algorithm has been trained with 8 A100 GPUs. I am having two instances, each equipped with 4 A100s instead of one GPU instance with 8 A100 GPUs. Is there any way to specify the instances in the configurations? In another word, where can I specify the number of nodes in the code? https://lightning.ai/docs/pytorch/stable/common/trainer.html#num-nodes

I would do appreciate if you could give a comment on these.

Update:

I added number of nodes to the training process and sent a pull request. In case of being accepted, this issue will be closed.

CheruscanArminius avatar Jan 05 '24 19:01 CheruscanArminius