Offline Datasets res net optionality

2020-03-12 18:32:23 +01:00
parent 2f99341cc3
commit bb47e07566
11 changed files with 638 additions and 140 deletions
--- a/lib/models/homotopy_classification/cnn_based.py
+++ b/lib/models/homotopy_classification/cnn_based.py
@ -60,7 +60,7 @@ class ConvHomDetector(LightningBaseModule):
        super(ConvHomDetector, self).__init__(hparams)

        # Dataset
-        self.dataset = TrajData(self.hparams.data_param.map_root, mode='all_in_map', )
+        self.dataset = TrajData(self.hparams.data_param.map_root, mode='classifier_all_in_map', )

        # Additional Attributes
        self.map_shape = self.dataset.map_shapes_max