We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 68bed59 commit 53f88d2Copy full SHA for 53f88d2
codes/model/model_interface.py
@@ -20,7 +20,7 @@
20
class MInterface(pl.LightningModule):
21
def __init__(self, model_name, loss, lr, **kargs):
22
super().__init__()
23
- self.save_hyperparameters()
+ self.save_hyperparameters(ignore=["callbacks"])
24
self.configure_loss()
25
self.model = self.load_model()
26
self.teacher_model = self.load_model()
0 commit comments