Calculates the modulus between the original batch size and number of currently assigned worker images then distributes among non-complementary jobs. Might behave weirdly for asymmetrically sized setups but tests fine on my 4x Tesla P40 setup.
Added original_batch_size as total_batch_size is overwritten later in the code which would impact the modulus math. I tried integrating it into the deferred images count so that slower gpus could take extra images but ran into issues. Closes #6
papuSpartan
commented
1 year ago
Calculates the modulus between the original batch size and number of currently assigned worker images then distributes among non-complementary jobs. Might behave weirdly for asymmetrically sized setups but tests fine on my 4x Tesla P40 setup.
Added original_batch_size as total_batch_size is overwritten later in the code which would impact the modulus math. I tried integrating it into the deferred images count so that slower gpus could take extra images but ran into issues. Closes #6