Update gitignore and fix bugs

hrishi508 · Jun 29, 2022 · 9b7bd4e · 9b7bd4e
1 parent 6c72f0f
commit 9b7bd4e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -19,3 +19,5 @@ __pycache__/
 datasets/
 weights/
 temp/
+logs.txt
+model.py
diff --git a/train.py b/train.py
@@ -242,7 +242,7 @@ def main(args):
 
     model = DebFace(cfg).to(cfg.device)
     # summary(model, (3, 112, 112))
-    print(cfg.load_weights)
+    # print(cfg.load_weights)
     if cfg.load_weights:
         model.load_state_dict(torch.load(cfg.model_weights_dir + cfg.load_weights_file))
 

diff --git a/train_without_race.py b/train_without_race.py
@@ -339,7 +339,7 @@ def main(args):
         test(test_loader, model, loss_fn_arr, test_loss_arr, cfg)
 
         # code used for testing model logic using random dataset created above
-        # train(dataloader, model, loss_fn_arr, train_loss_arr, optimizer, cfg)
+        # train(dataloader, model, loss_fn_arr, train_loss_arr, optimizer, scheduler, cfg)
         # test(dataloader, model, loss_fn_arr, test_loss_arr, cfg)
 
         if cfg.save_model_weights_every > 0 and (t + 1)%cfg.save_model_weights_every == 0: