From 162ddcf6c717e0362fd76b731e2501b967ec5930 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 27 Dec 2019 13:04:46 -0800
Subject: [PATCH] updates

---
 train.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/train.py b/train.py
index 3e903c03..7a9d2d07 100644
--- a/train.py
+++ b/train.py
@@ -403,15 +403,15 @@ def train():
 def prebias():
     # trains output bias layers for 1 epoch and creates new backbone
     if opt.prebias:
-        a = opt.img_weights  # save settings
-        opt.img_weights = False  # disable settings
+        # opt_0 = opt  # save settings
+        # opt.rect = False  # update settings (if any)
 
         train()  # transfer-learn yolo biases for 1 epoch
         create_backbone(last)  # saved results as backbone.pt
 
+        # opt = opt_0  # reset settings
         opt.weights = wdir + 'backbone.pt'  # assign backbone
         opt.prebias = False  # disable prebias
-        opt.img_weights = a  # reset settings
 
 
 if __name__ == '__main__':