god I need to replace omegaconf

This commit is contained in:
mrq 2024-05-12 14:01:52 -05:00
parent 5eb5db7f7f
commit 8d79f78e0a
2 changed files with 21 additions and 0 deletions

Binary file not shown.

View File

@ -654,6 +654,27 @@ class Config(_Config):
self.dataset.use_hdf5 = False
def format( self ):
if isinstance(self.dataset, type):
self.dataset = dict()
if isinstance(self.model, type):
self.model = dict()
if isinstance(self.hyperparameters, type):
self.hyperparameters = dict()
if isinstance(self.evaluation, type):
self.evaluation = dict()
if isinstance(self.trainer, type):
self.trainer = dict()
if isinstance(self.inference, type):
self.inference = dict()
if isinstance(self.optimizations, type):
self.optimizations = dict()
self.dataset = Dataset(**self.dataset)
self.dataset.training = [ Path(dir) for dir in self.dataset.training ]
self.dataset.validation = [ Path(dir) for dir in self.dataset.validation ]