Add `batch_size` to create_dataloader

open-forest-observatory / tree-detection-framework

BSD 3-Clause "New" or "Revised" License

3 stars 0 forks source link

Add `batch_size` to create_dataloader #20

Closed russelldj closed 1 month ago

russelldj commented 1 month ago

@youngdjn mentioned that it would be helpful to control the batch size, especially when we want to fully utilize the GPU. This should be a pretty easy addition to create_data_loader but may require some attention downstream to properly handle batches of length greater than one.

amrithasp02 commented 1 month ago

Completed in pull request #31