fix: dont save component initializer as hparm

This commit is contained in:
Alexander Engelsberger 2022-06-12 11:40:33 +02:00
parent 60d2a1d2c9
commit 1be7d7ec09

View File

@ -33,7 +33,10 @@ class BaseYArchitecture(pl.LightningModule):
del hparams["initialized_proto_shape"]
hparams = self.HyperParameters(**hparams)
else:
self.save_hyperparameters(hparams.__dict__)
self.save_hyperparameters(
hparams.__dict__,
ignore=["component_initializer"],
)
super().__init__()
@ -217,7 +220,6 @@ class BaseYArchitecture(pl.LightningModule):
self.update_metrics_epoch()
def on_save_checkpoint(self, checkpoint: dict[str, Any]) -> None:
checkpoint["hyper_parameters"]["component_initializer"] = None
checkpoint["hyper_parameters"] = {
'hparams': checkpoint["hyper_parameters"]
}