torchgeo
torchgeo copied to clipboard
OSCDDataModule initialises with batch_size 1, ignoring the configured batch_size
Description
super().__init__(OSCD, 1, num_workers, **kwargs)
However other datamodules pass along the batchsize: e.g.
super().__init__(UCMerced, batch_size, num_workers, **kwargs)
This also affects:
- https://github.com/microsoft/torchgeo/blob/main/torchgeo/datamodules/levircd.py
Steps to reproduce
Use the datamodule
Version
dev