diff --git a/main_linear.py b/main_linear.py
index a5c179a..14e403e 100644
--- a/main_linear.py
+++ b/main_linear.py
@@ -246,7 +246,7 @@ def main(args):
                 'best_acc1': best_acc1,
                 'optimizer' : optimizer.state_dict(),
             }, is_best, args.output_dir)
-            if epoch == args.start_epoch:
+            if epoch + 1 == args.start_epoch + args.print_freq:
                 sanity_check(model.state_dict(), args.pretrained, linear_keyword, visual_keyword)
 
         log_stats = {**{f'train_{k}': v for k, v in train_stats.items()},