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.
2 parents 9ef548d + f99530c commit a9e7027Copy full SHA for a9e7027
cebra/solver/base.py
@@ -231,7 +231,8 @@ def fit(self,
231
self.decoding(loader, valid_loader))
232
if save_hook is not None:
233
save_hook(num_steps, self)
234
- self.save(logdir, f"checkpoint_{num_steps:#07d}.pth")
+ if logdir is not None:
235
+ self.save(logdir, f"checkpoint_{num_steps:#07d}.pth")
236
237
def step(self, batch: cebra.data.Batch) -> dict:
238
"""Perform a single gradient update.
0 commit comments