Closed lchia closed 3 years ago
IndexError in DataLoader worker process 7...
This is the script of training:
CUDA_VISIBLE_DEVICES=0,1 \ python main_refinement_flow.py \ --batch_size 3\ --gpus 2 \ --threads 30
I faced the same issue while trying training on Dataset of length 785 with corresponding Dataloader with batch size of 8. Making Dataset length divisible by the batch size solved the issue
Updated training script, works better.
CUDA_VISIBLE_DEVICES=0,1,2 \ python main_refinement_flow.py \ --batch_size 3\ --gpus 3 \ --threads 30
IndexError in DataLoader worker process 7...
This is the script of training:
I faced the same issue while trying training on Dataset of length 785 with corresponding Dataloader with batch size of 8. Making Dataset length divisible by the batch size solved the issue
Updated training script, works better.