Skip to content

Conversation

abhi-mosaic
Copy link
Contributor

@abhi-mosaic abhi-mosaic commented Mar 3, 2022

This PR is a quality-of-life improvement for training with the C4 dataset, instead of erroring out and killing the training job, it just overwrites dataloader.num_workers before initializing the dataloader.

I expect to relax the num_workers=1 constraint in the future, but for now it just makes sharding simpler.

@abhi-mosaic abhi-mosaic requested a review from siriuslee March 3, 2022 01:57
@abhi-mosaic abhi-mosaic self-assigned this Mar 3, 2022
Copy link
Contributor

@siriuslee siriuslee left a comment

Choose a reason for hiding this comment

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

LGTM

@abhi-mosaic abhi-mosaic merged commit ab1854d into dev Mar 3, 2022
@abhi-mosaic abhi-mosaic deleted the abhi/c4_bugfix branch March 3, 2022 03:42
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.

2 participants