steffen/hom_traj_gen
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Debugging Validation and testing

Browse Source
This commit is contained in:
Si11ium
2020-03-09 19:18:22 +01:00
parent 4ae333fe5d
commit 6b9696c98e
14 changed files with 28 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
lib/modules/utils.py
View File

@@ -90,7 +90,7 @@ class LightningBaseModule(pl.LightningModule, ABC):
# Data loading
# =============================================================================
# Map Object
self.map_storage = MapStorage(self.hparams.data_param.map_root)
self.map_storage = MapStorage(self.hparams.data_param.map_root, load_all=True)
def size(self):
return self.shape
@@ -143,19 +143,19 @@ class LightningBaseModule(pl.LightningModule, ABC):
# Train Dataloader
def train_dataloader(self):
return DataLoader(dataset=self.dataset.train_dataset, shuffle=True,
batch_size=self.hparams.data_param.batchsize,
batch_size=self.hparams.train_param.batch_size,
num_workers=self.hparams.data_param.worker)
# Test Dataloader
def test_dataloader(self):
return DataLoader(dataset=self.dataset.test_dataset, shuffle=True,
batch_size=self.hparams.data_param.batchsize,
batch_size=self.hparams.train_param.batch_size,
num_workers=self.hparams.data_param.worker)
# Validation Dataloader
def val_dataloader(self):
return DataLoader(dataset=self.dataset.val_dataset, shuffle=False,
batch_size=self.hparams.data_param.batchsize,
batch_size=self.hparams.train_param.batch_size,
num_workers=self.hparams.data_param.worker)