Skip to content

[Fix] Fix bug in non-distributed multi-gpu training/testing#7019

Merged
ZwwWayne merged 2 commits intoopen-mmlab:devfrom
AronLin:gpu
Jan 26, 2022
Merged

[Fix] Fix bug in non-distributed multi-gpu training/testing#7019
ZwwWayne merged 2 commits intoopen-mmlab:devfrom
AronLin:gpu

Conversation

@AronLin
Copy link
Copy Markdown
Contributor

@AronLin AronLin commented Jan 17, 2022

Since MMDP does not support non-distributed multi-GPU training, --gpus in train.py lost its role, so it is removed, --gpu-ids is changed to --gpu-id, because only one GPU can be specified for non-distribution training and testing.
If the number of GPUs is more than 1, there will be an assertion error in MMDP because MMDP does not support the situation with more than one GPU.

Copy link
Copy Markdown
Collaborator

@BIGWangYuDong BIGWangYuDong left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

'--gpus',
'--gpu-id',
type=int,
help='number of gpus to use '
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We should add deprecation warning of --gpu-ids and --gpus to indicate users that they are deprecated and useless. We should not delete them immediately.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

ok

Copy link
Copy Markdown
Contributor

@gaotongxiao gaotongxiao left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Never mind... It should have been posted to mmocr :)

@AronLin AronLin deleted the gpu branch February 14, 2022 21:29
chhluo pushed a commit to chhluo/mmdetection that referenced this pull request Feb 21, 2022
…ab#7019)

* Fix bug in non-distributed training/testing

* Add deprecated message
ZwwWayne pushed a commit that referenced this pull request Jul 18, 2022
* Fix bug in non-distributed training/testing

* Add deprecated message
ZwwWayne pushed a commit to ZwwWayne/mmdetection that referenced this pull request Jul 19, 2022
…ab#7019)

* Fix bug in non-distributed training/testing

* Add deprecated message
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants