Skip to content
Snippets Groups Projects
Commit 1e432168 authored by Michael Langguth's avatar Michael Langguth
Browse files

Merge branch 'develop' into michael_issue#117_enforce_CPU_parallelization_preprocessing

parents 5c3bb92d d6773ba6
Branches
Tags
No related merge requests found
Pipeline #70696 passed
...@@ -3,9 +3,9 @@ ...@@ -3,9 +3,9 @@
"batch_size": 4, "batch_size": 4,
"lr": 0.001, "lr": 0.001,
"max_epochs":20, "max_epochs":20,
"context_frames":10, "context_frames":12,
"loss_fun":"rmse", "loss_fun":"rmse",
"shuffle_on_val":false "shuffle_on_val":true
} }
......
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
"batch_size": 10, "batch_size": 10,
"lr": 0.001, "lr": 0.001,
"max_epochs": 2, "max_epochs": 2,
"context_frames":10 "context_frames": 12
} }
......
...@@ -11,5 +11,5 @@ ...@@ -11,5 +11,5 @@
"state_weight": 0.0, "state_weight": 0.0,
"nz": 32, "nz": 32,
"max_epochs":2, "max_epochs":2,
"context_frames":10 "context_frames":12
} }
...@@ -13,7 +13,7 @@ ...@@ -13,7 +13,7 @@
"state_weight": 0.0, "state_weight": 0.0,
"nz": 16, "nz": 16,
"max_epochs":2, "max_epochs":2,
"context_frames":10 "context_frames": 12
} }
...@@ -4,10 +4,10 @@ ...@@ -4,10 +4,10 @@
"lr": 0.001, "lr": 0.001,
"nz":16, "nz":16,
"max_epochs":2, "max_epochs":2,
"context_frames":10, "context_frames":12,
"weight_recon":1, "weight_recon":1,
"loss_fun": "rmse", "loss_fun": "rmse",
"shuffle_on_val": false "shuffle_on_val": true
} }
......
...@@ -139,11 +139,12 @@ class TrainModel(object): ...@@ -139,11 +139,12 @@ class TrainModel(object):
Setup train and val dataset instance with the corresponding data split configuration. Setup train and val dataset instance with the corresponding data split configuration.
Simultaneously, sequence_length is attached to the hyperparameter dictionary. Simultaneously, sequence_length is attached to the hyperparameter dictionary.
""" """
VideoDataset = datasets.get_dataset_class(self.dataset) self.train_dataset = VideoDataset(input_dir=self.input_dir, mode='train', datasplit_config=self.datasplit_dict,
self.train_dataset = VideoDataset(input_dir=self.input_dir,mode='train',datasplit_config=self.datasplit_dict) hparams_dict_config=self.model_hparams_dict)
self.val_dataset = VideoDataset(input_dir=self.input_dir, mode='val',datasplit_config=self.datasplit_dict) self.val_dataset = VideoDataset(input_dir=self.input_dir, mode='val', datasplit_config=self.datasplit_dict,
hparams_dict_config=self.model_hparams_dict)
self.model_hparams_dict_load.update({"sequence_length": self.train_dataset.sequence_length}) # ML/BG 2021-06-15: Is the following needed?
# self.model_hparams_dict_load.update({"sequence_length": self.train_dataset.sequence_length})
def setup_model(self): def setup_model(self):
""" """
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment