From fe896c17925397072746f47a5f6a9af0dc6b3b31 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 5 Apr 2019 15:43:41 +0200
Subject: [PATCH] updates

---
 train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/train.py b/train.py
index b6dd03ad..a62612ce 100644
--- a/train.py
+++ b/train.py
@@ -190,7 +190,7 @@ def train(
             best_loss = results[0]
 
         # Save training results
-        save = True and not opt.no_save
+        save = True and not opt.nosave
         if save:
             # Create checkpoint
             chkpt = {'epoch': epoch,
@@ -230,7 +230,7 @@ if __name__ == '__main__':
     parser.add_argument('--rank', default=0, type=int, help='distributed training node rank')
     parser.add_argument('--world-size', default=1, type=int, help='number of nodes for distributed training')
     parser.add_argument('--backend', default='nccl', type=str, help='distributed backend')
-    parser.add_argument('--no-save', action='store_false', help='transfer learning flag')
+    parser.add_argument('--nosave', action='store_true', help='do not save training results')
     opt = parser.parse_args()
     print(opt, end='\n\n')