You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Thank you very much for your work.
I'm trying to train my model based on my own dataset, but when running the training command it always fails with the following error:
File "/home/ubuntu/git/stylegan-v/env-ampere/lib/python3.9/site-packages/torch/utils/data/sampler.py", line 215, in __init__
raise ValueError("batch_size should be a positive integer value, "
ValueError: batch_size should be a positive integer value, but got batch_size=0
Even if I try to specify the batch size in the command by adding training.batch_size=8 for instance, it keeps failing with the same error message.
So, I realised that I have this issue when I use my dataset with a 512 resolution.
If I use the exact same dataset but with a resolution of 256, it works.
It makes no sense to me. I printed out the value of batch_size in the sampler.py file and the value is changing at each training iteration.
For some reason it works at a resolution of 256, but when the resolution is at 512, the batch_size turns to 0.
I'm specifying the training.batch_size value when running the training command, and it is correctly written in the experiment_confog.yaml file.
I'm at a complete loss as to how I could train my model with a dataset at a 512 resolution.
Would anyone have any idea please ?
Hello guys,
Thank you very much for your work.
I'm trying to train my model based on my own dataset, but when running the training command it always fails with the following error:
Even if I try to specify the batch size in the command by adding
training.batch_size=8
for instance, it keeps failing with the same error message.The command I use is:
Would someone have any idea what I should do to get around this please?
Thank you very much for your help in advance.
Cheers
The text was updated successfully, but these errors were encountered: