train running

2020-03-09 21:41:50 +01:00
parent daed810958
commit 6cc978e464
7 changed files with 68 additions and 57 deletions
--- a/lib/modules/utils.py
+++ b/lib/modules/utils.py
@@ -90,7 +90,7 @@ class LightningBaseModule(pl.LightningModule, ABC):
        # Data loading
        # =============================================================================
        # Map Object
-        self.map_storage = MapStorage(self.hparams.data_param.map_root, load_all=True)
+        self.map_storage = MapStorage(self.hparams.data_param.map_root)

    def size(self):
        return self.shape