torchgeo icon indicating copy to clipboard operation
torchgeo copied to clipboard

OSCDDataModule initialises with batch_size 1, ignoring the configured batch_size

Open robmarkcole opened this issue 1 year ago • 0 comments

Description

In the init of OSCD:

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

robmarkcole avatar Feb 19 '24 13:02 robmarkcole