diff --git a/src/frdc/train/frdc_datamodule.py b/src/frdc/train/frdc_datamodule.py index 97daaa6b..9f35ce03 100644 --- a/src/frdc/train/frdc_datamodule.py +++ b/src/frdc/train/frdc_datamodule.py @@ -36,7 +36,7 @@ class FRDCDataModule(LightningDataModule): Does not have the same performance as:: - FRDCSSLDataModule( + FRDCDataModule( train_lab_ds=train_lab_ds, train_unl_ds=None, ...