From 288745a100d2026e0045ba86cf955ee27270fb31 Mon Sep 17 00:00:00 2001
From: Bhautik Donga <bhautikdonga2000@gmail.com>
Date: Mon, 31 Jan 2022 18:33:17 +0530
Subject: [PATCH] Update train.py

Resolved typos
---
 train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/train.py b/train.py
index 1eda27a4..c0690d73 100644
--- a/train.py
+++ b/train.py
@@ -448,7 +448,7 @@ def burnin_schedule(i):
                     pass
                 save_path = os.path.join(config.checkpoints, f'{save_prefix}{epoch + 1}.pth')
                 if isinstance(model, torch.nn.DataParallel):
-                    torch.save(model.moduel,state_dict(), save_path)
+                    torch.save(model.module.state_dict(), save_path)
                 else:
                     torch.save(model.state_dict(), save_path)
                 logging.info(f'Checkpoint {epoch + 1} saved !')