-
Notifications
You must be signed in to change notification settings - Fork 160
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Enable miniepoch for MultiProcessingReadingService #1170
Conversation
This pull request was exported from Phabricator. Differential Revision: D45666912 |
…ch#1170) Summary: Pull Request resolved: pytorch#1170 A retry of pytorch#1147 - Add pause, resume, limit API to MultiProcessingReadingService to support miniepoch API. - When limit is requested, `IterQueueWrapper` will make sure only `limit` number of requests are sent to worker process and dispatching proccess. - Consolidate a few API/messages - reset and reset_epoch is merged as a single message - Process reset function is shared for both worker process and dispatching process - For pause/resume/limit/reset_epoch, add a counter to sync between loops to make sure the DataPipe graph is in-place modified once. - Remove unused thread eventloop Differential Revision: D45666912 fbshipit-source-id: 48176f10354494754f47a96a1ca166ed45e7e247
This pull request was exported from Phabricator. Differential Revision: D45666912 |
…ch#1170) Summary: Pull Request resolved: pytorch#1170 A retry of pytorch#1147 - Add pause, resume, limit API to MultiProcessingReadingService to support miniepoch API. - When limit is requested, `IterQueueWrapper` will make sure only `limit` number of requests are sent to worker process and dispatching proccess. - Consolidate a few API/messages - reset and reset_epoch is merged as a single message - Process reset function is shared for both worker process and dispatching process - For pause/resume/limit/reset_epoch, add a counter to sync between loops to make sure the DataPipe graph is in-place modified once. - Remove unused thread eventloop Differential Revision: D45666912 fbshipit-source-id: a2e064371a510463cc13253419aa7d95233bf41e
This pull request was exported from Phabricator. Differential Revision: D45666912 |
@ejguan has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
Summary: A retry of pytorch#1147 - Add pause, resume, limit API to MultiProcessingReadingService to support miniepoch API. - When limit is requested, `IterQueueWrapper` will make sure only `limit` number of requests are sent to worker process and dispatching proccess. - Consolidate a few API/messages - reset and reset_epoch is merged as a single message - Process reset function is shared for both worker process and dispatching process - For pause/resume/limit/reset_epoch, add a counter to sync between loops to make sure the DataPipe graph is in-place modified once. - Remove unused thread eventloop Pull Request resolved: pytorch#1170 Differential Revision: D45666912 Pulled By: ejguan fbshipit-source-id: 1e88dde3af0f406c062ca523e7b96a3c79a74b6d
This pull request was exported from Phabricator. Differential Revision: D45666912 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
A few questions/notes. Overall, it does look good but our codebase is getting very complicated.
Summary: A retry of pytorch#1147 - Add pause, resume, limit API to MultiProcessingReadingService to support miniepoch API. - When limit is requested, `IterQueueWrapper` will make sure only `limit` number of requests are sent to worker process and dispatching proccess. - Consolidate a few API/messages - reset and reset_epoch is merged as a single message - Process reset function is shared for both worker process and dispatching process - For pause/resume/limit/reset_epoch, add a counter to sync between loops to make sure the DataPipe graph is in-place modified once. - Remove unused thread eventloop Pull Request resolved: pytorch#1170 Differential Revision: D45666912 Pulled By: ejguan fbshipit-source-id: 1e88dde3af0f406c062ca523e7b96a3c79a74b6d
@ejguan has imported this pull request. If you are a Meta employee, you can view this diff on Phabricator. |
Summary: A retry of pytorch#1147 - Add pause, resume, limit API to MultiProcessingReadingService to support miniepoch API. - When limit is requested, `IterQueueWrapper` will make sure only `limit` number of requests are sent to worker process and dispatching proccess. - Consolidate a few API/messages - reset and reset_epoch is merged as a single message - Process reset function is shared for both worker process and dispatching process - For pause/resume/limit/reset_epoch, add a counter to sync between loops to make sure the DataPipe graph is in-place modified once. - Remove unused thread eventloop Pull Request resolved: pytorch#1170 Differential Revision: D45666912 Pulled By: ejguan fbshipit-source-id: ddff582f35a25528a2b8a1751ea0d0359f866ed0
This pull request was exported from Phabricator. Differential Revision: D45666912 |
Summary: A retry of pytorch#1147 - Add pause, resume, limit API to MultiProcessingReadingService to support miniepoch API. - When limit is requested, `IterQueueWrapper` will make sure only `limit` number of requests are sent to worker process and dispatching proccess. - Consolidate a few API/messages - reset and reset_epoch is merged as a single message - Process reset function is shared for both worker process and dispatching process - For pause/resume/limit/reset_epoch, add a counter to sync between loops to make sure the DataPipe graph is in-place modified once. - Remove unused thread eventloop Pull Request resolved: pytorch#1170 Differential Revision: D45666912 Pulled By: ejguan fbshipit-source-id: f533892a1ffcdc103988a724bea8ce20572465b6
This pull request was exported from Phabricator. Differential Revision: D45666912 |
Summary: A retry of pytorch#1147 - Add pause, resume, limit API to MultiProcessingReadingService to support miniepoch API. - When limit is requested, `IterQueueWrapper` will make sure only `limit` number of requests are sent to worker process and dispatching proccess. - Consolidate a few API/messages - reset and reset_epoch is merged as a single message - Process reset function is shared for both worker process and dispatching process - For pause/resume/limit/reset_epoch, add a counter to sync between loops to make sure the DataPipe graph is in-place modified once. - Remove unused thread eventloop Pull Request resolved: pytorch#1170 Differential Revision: D45666912 Pulled By: ejguan fbshipit-source-id: 40ec4eef0fdf1de05d3a545e08c11f155446ae2d
This pull request was exported from Phabricator. Differential Revision: D45666912 |
Summary: A retry of pytorch#1147 - Add pause, resume, limit API to MultiProcessingReadingService to support miniepoch API. - When limit is requested, `IterQueueWrapper` will make sure only `limit` number of requests are sent to worker process and dispatching proccess. - Consolidate a few API/messages - reset and reset_epoch is merged as a single message - Process reset function is shared for both worker process and dispatching process - For pause/resume/limit/reset_epoch, add a counter to sync between loops to make sure the DataPipe graph is in-place modified once. - Remove unused thread eventloop Pull Request resolved: pytorch#1170 Differential Revision: D45666912 Pulled By: ejguan fbshipit-source-id: ffafcbb0b78add9c863d2faaec7248332ef36f68
This pull request was exported from Phabricator. Differential Revision: D45666912 |
Summary: A retry of pytorch#1147 - Add pause, resume, limit API to MultiProcessingReadingService to support miniepoch API. - When limit is requested, `IterQueueWrapper` will make sure only `limit` number of requests are sent to worker process and dispatching proccess. - Consolidate a few API/messages - reset and reset_epoch is merged as a single message - Process reset function is shared for both worker process and dispatching process - For pause/resume/limit/reset_epoch, add a counter to sync between loops to make sure the DataPipe graph is in-place modified once. - Remove unused thread eventloop Pull Request resolved: pytorch#1170 Differential Revision: D45666912 Pulled By: ejguan fbshipit-source-id: 7f4d208e3ff0a6bea4688862f7b8d5456af8e7f6
This pull request was exported from Phabricator. Differential Revision: D45666912 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM! Thanks!
Summary:
A retry of #1147
IterQueueWrapper
will make sure onlylimit
number of requests are sent to worker process and dispatching proccess.Differential Revision: D45666912