search

knoriy / CLARA

Apache License 2.0
61 stars 3 forks source link

Effective batch size does not work #3

Closed knoriy closed 2 years ago

knoriy commented 2 years ago

PyTorch lighting handels the effective batch size. We simply need to pass the batch size. PL will calculate batch_size * n_gpu * n_node