From 059fa85b45ad049b89179a8f945479d7bc27f42b Mon Sep 17 00:00:00 2001
From: Anthony Ortiz <anthony.ortiz@microsoft.com>
Date: Mon, 4 Oct 2021 01:36:06 +0000
Subject: [PATCH] byol

---
 train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train.py b/train.py
index 6cb51dea1..cb27f8536 100755
--- 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),