Skip to content

Commit f99530c

Browse files
authored
Update base.py (#224)
This is a lazy solution to #223
1 parent c32ed67 commit f99530c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

cebra/solver/base.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,8 @@ def fit(
210210
self.decoding(loader, valid_loader))
211211
if save_hook is not None:
212212
save_hook(num_steps, self)
213-
self.save(logdir, f"checkpoint_{num_steps:#07d}.pth")
213+
if logdir is not None:
214+
self.save(logdir, f"checkpoint_{num_steps:#07d}.pth")
214215

215216
def step(self, batch: cebra.data.Batch) -> dict:
216217
"""Perform a single gradient update.

0 commit comments

Comments
 (0)