We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
In the example of deeplab training in the yaml config file deeplab/yamls/deeplabv3.yaml it is written in config file:
yaml
batch_size: 128 # Training dataloader batch size per device - line 26 batch_size: 128 # Evaluation dataloader batch size per device - line 39
But, according to the main.py (lines 64-66) it should be not the batch size per device rather the total batch size.
main.py
batch size per device
total batch size
if dist.get_world_size(): train_batch_size //= dist.get_world_size() eval_batch_size //= dist.get_world_size()
The text was updated successfully, but these errors were encountered:
Yes, that comment is incorrect. batch_size is the total batch size not per device batch size. We will fix it ASAP, thank you for the notice!
batch_size
Sorry, something went wrong.
A-Jacobson
No branches or pull requests
In the example of deeplab training in the
yaml
config file deeplab/yamls/deeplabv3.yaml it is written in config file:But, according to the
main.py
(lines 64-66) it should be not thebatch size per device
rather thetotal batch size
.The text was updated successfully, but these errors were encountered: