From 37fc64e309eb4a21f46f420ea8a0ec0fd24823f8 Mon Sep 17 00:00:00 2001
From: Maxwell Aladago
 <aladago@vpn-investment-office-231-129-233.dartmouth.edu>
Date: Fri, 25 Feb 2022 15:28:55 -0500
Subject: [PATCH] Improvement: Ensure sanity check happens when starting linear
 training from scarch

---
 main_linear.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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()},