Skip to content
This repository was archived by the owner on Apr 17, 2023. It is now read-only.

[OTX] Enable multi-GPU training#93

Merged
eunwoosh merged 17 commits intootxfrom
multigpu_enablement
Dec 20, 2022
Merged

[OTX] Enable multi-GPU training#93
eunwoosh merged 17 commits intootxfrom
multigpu_enablement

Conversation

@eunwoosh
Copy link
Copy Markdown
Contributor

@eunwoosh eunwoosh commented Nov 25, 2022

Summary

Enable multi-GPU training for classification, detection, segmentation tasks.

This PR includes

  • enable multi GPU training
  • combine train_worker() into run() in each task
  • modify some samplers to run on multi GPU training

@eunwoosh eunwoosh requested a review from supersoob November 25, 2022 12:13
@eunwoosh eunwoosh force-pushed the multigpu_enablement branch from e53bfa7 to 8300627 Compare December 8, 2022 06:50
@eunwoosh eunwoosh changed the title Multigpu enablement [OTX] Enable multi-GPU training Dec 15, 2022
@eunwoosh eunwoosh force-pushed the multigpu_enablement branch 2 times, most recently from dfa6048 to 4700542 Compare December 15, 2022 07:45
@eunwoosh eunwoosh marked this pull request as ready for review December 15, 2022 07:45
Copy link
Copy Markdown
Contributor

@harimkang harimkang left a comment

Choose a reason for hiding this comment

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

add minor comments

@eunwoosh eunwoosh force-pushed the multigpu_enablement branch from 4700542 to 255ac6f Compare December 20, 2022 01:44
Copy link
Copy Markdown

@supersoob supersoob left a comment

Choose a reason for hiding this comment

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

LGTM.

@eunwoosh eunwoosh merged commit 0322785 into otx Dec 20, 2022
@eunwoosh eunwoosh deleted the multigpu_enablement branch December 20, 2022 02:08
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants