diff options
Diffstat (limited to 'trainer/base.py')
-rw-r--r-- | trainer/base.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/trainer/base.py b/trainer/base.py index e700dd6..1f85e71 100644 --- a/trainer/base.py +++ b/trainer/base.py | |||
@@ -74,7 +74,7 @@ class Checkpointer(): | |||
74 | def checkpoint(self, step: int, postfix: str): | 74 | def checkpoint(self, step: int, postfix: str): |
75 | pass | 75 | pass |
76 | 76 | ||
77 | @torch.inference_mode() | 77 | @torch.no_grad() |
78 | def save_samples(self, step: int): | 78 | def save_samples(self, step: int): |
79 | print(f"Saving samples for step {step}...") | 79 | print(f"Saving samples for step {step}...") |
80 | 80 | ||