julius/prototorch_models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: dont save component initializer as hparm

Browse Source
This commit is contained in:
Alexander Engelsberger 2022-06-12 11:40:33 +02:00
parent 60d2a1d2c9
commit 1be7d7ec09
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
prototorch/y/architectures/base.py
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"]
}