Figure plotting with Eurosat & classification trainer gives blank images

[robmarkcole]

Description

As title:

Steps to reproduce

datamodule = EuroSATDataModule(batch_size=64, num_workers=8, bands=EuroSAT.BAND_SETS["all"])
datamodule.setup(stage="fit")

task = ClassificationTask(
    model="resnet18",
    in_channels=13,
    num_classes=10,
    loss="ce",
    lr=0.001,
)

logger = TensorBoardLogger("tb_logs", name="my_eurosat")

trainer = L.Trainer(
    max_epochs=10,
    logger=logger
)
trainer.fit(task, datamodule)

Version

master

[robmarkcole]

Appears the issue is not with the basic plotting:

[adamjstewart]

I'm guessing this is related to #1263? That is, dataset plotting works, but data module plotting (in which data augmentation has been applied) does not? The solution for this dataset is the same as for all other datasets where this issue occurs, and is described in #1263.

[robmarkcole]

Noted this is likely effecting many datasets and due to the norm. Closing and continuing on the linked issue