Skip to content

Commit 53f88d2

Browse files
authored
Update model_interface.py
1 parent 68bed59 commit 53f88d2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

codes/model/model_interface.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
class MInterface(pl.LightningModule):
2121
def __init__(self, model_name, loss, lr, **kargs):
2222
super().__init__()
23-
self.save_hyperparameters()
23+
self.save_hyperparameters(ignore=["callbacks"])
2424
self.configure_loss()
2525
self.model = self.load_model()
2626
self.teacher_model = self.load_model()

0 commit comments

Comments
 (0)