From be934ba5a5f92087bf729a538ce0f081f06791c3 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 9 Feb 2019 19:26:53 +0100
Subject: [PATCH] updates

---
 train.py | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/train.py b/train.py
index e9039d42..57aabca4 100644
--- a/train.py
+++ b/train.py
@@ -196,13 +196,7 @@ def train(
             os.system('cp ' + latest_weights_file + ' ' + os.path.join(weights, 'backup{}.pt'.format(epoch)))
 
         # Calculate mAP
-        mAP, R, P = test.test(
-            cfg,
-            data_cfg,
-            latest_weights_file,
-            batch_size=batch_size,
-            img_size=img_size,
-        )
+        mAP, R, P = test.test(cfg, data_cfg, weights=latest_weights_file, batch_size=batch_size, img_size=img_size)
 
         # Write epoch results
         with open('results.txt', 'a') as file:
@@ -228,7 +222,6 @@ if __name__ == '__main__':
 
     init_seeds()
 
-    torch.cuda.empty_cache()
     train(
         opt.cfg,
         opt.data_cfg,