We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c32ed67 commit f99530cCopy full SHA for f99530c
cebra/solver/base.py
@@ -210,7 +210,8 @@ def fit(
210
self.decoding(loader, valid_loader))
211
if save_hook is not None:
212
save_hook(num_steps, self)
213
- self.save(logdir, f"checkpoint_{num_steps:#07d}.pth")
+ if logdir is not None:
214
+ self.save(logdir, f"checkpoint_{num_steps:#07d}.pth")
215
216
def step(self, batch: cebra.data.Batch) -> dict:
217
"""Perform a single gradient update.
0 commit comments