[Bug: 865] datamodule.setup() assertion failed (#1058)

* [Bug: 839] Crop in SSPCAB implementation
This fix follows the official implementation for the SSPCAB by the original author

Fixes #839

Signed-off-by: Ng, Kin Nam <kin.nam.ng@intel.com>
Signed-off-by: Lim, Kuan Xian <kuan.xian.lim@intel.com>

* [Bug: 865] datamodule.setup() assertion failed

This fix download the dataset first before splitting the train/val/test sets.

Fixes #865

Signed-off-by: Ng, Kin Nam <kin.nam.ng@intel.com>
Signed-off-by: Lim, Kuan Xian <kuan.xian.lim@intel.com>

* [Bug: 839] Crop in SSPCAB implementation
This fix follows the official implementation for the SSPCAB by the original author

Fixes #839

Signed-off-by: Ng, Kin Nam <kin.nam.ng@intel.com>
Signed-off-by: Lim, Kuan Xian <kuan.xian.lim@intel.com>

remove trailing whitespace

---------

Signed-off-by: Ng, Kin Nam <kin.nam.ng@intel.com>
Signed-off-by: Lim, Kuan Xian <kuan.xian.lim@intel.com>
Co-authored-by: Samet Akcay <samet.akcay@intel.com>

notebooks/000_getting_started/001_getting_started.ipynb

@@ -229,8 +229,8 @@
    ],
    "source": [
     "datamodule = get_datamodule(config)\n",
-    "datamodule.setup()  # Downloads the dataset if it's not in the specified `root` directory\n",
-    "datamodule.prepare_data()  # Create train/val/test/prediction sets.\n",
+    "datamodule.prepare_data()  # Downloads the dataset if it's not in the specified `root` directory\n",
+    "datamodule.setup()  # Create train/val/test/prediction sets.\n",
     "\n",
     "i, data = next(enumerate(datamodule.val_dataloader()))\n",
     "print(data.keys())"