byol

2021-10-04 01:36:06 +00:00 · 2021-10-04 01:36:06 +00:00 · 059fa85b45
--- a/train.py
+++ b/train.py
@ -36,6 +36,7 @@ TASK_TO_MODULES_MAPPING: Dict[
 ] = {
    "byol": (BYOLTask, So2SatDataModule),
    "byol_landcoverai": (BYOLTask, LandcoverAIDataModule),
+    "byol_chesapeake_cvpr": (BYOLTask, ChesapeakeCVPRDataModule),
    "chesapeake_cvpr": (ChesapeakeCVPRSegmentationTask, ChesapeakeCVPRDataModule),
    "cyclone": (CycloneSimpleRegressionTask, CycloneDataModule),
    "landcoverai": (LandcoverAISegmentationTask, LandcoverAIDataModule),