fix: ignore prototype_win_ratios
by loading with strict=False
This commit is contained in:
parent
197b728c63
commit
15e7232747
@ -59,5 +59,7 @@ if __name__ == "__main__":
|
||||
|
||||
# Load saved model
|
||||
new_model = pt.models.GLVQ.load_from_checkpoint(
|
||||
checkpoint_path="./glvq_iris.ckpt")
|
||||
checkpoint_path="./glvq_iris.ckpt",
|
||||
strict=False,
|
||||
)
|
||||
print(new_model)
|
||||
|
@ -39,9 +39,9 @@ class GLVQ(SupervisedPrototypeModel):
|
||||
beta=self.hparams.transfer_beta,
|
||||
)
|
||||
|
||||
def on_save_checkpoint(self, checkpoint):
|
||||
if "prototype_win_ratios" in checkpoint["state_dict"]:
|
||||
del checkpoint["state_dict"]["prototype_win_ratios"]
|
||||
# def on_save_checkpoint(self, checkpoint):
|
||||
# if "prototype_win_ratios" in checkpoint["state_dict"]:
|
||||
# del checkpoint["state_dict"]["prototype_win_ratios"]
|
||||
|
||||
def initialize_prototype_win_ratios(self):
|
||||
self.register_buffer(
|
||||
|
Loading…
Reference in New Issue
Block a user