Update train.py

2022-08-23 20:46:36 -04:00
parent f106897120
commit c9371bc72b
1 changed files with 2 additions and 2 deletions
--- a/keras-yolo3-master/train.py
+++ b/keras-yolo3-master/train.py
@@ -274,8 +274,8 @@ def _main_(args):


    # make a GPU version of infer_model for evaluation
-    #if multi_gpu > 1:
-    #    infer_model = load_model(config['train']['saved_weights_name'])
+    if multi_gpu > 1:
+        infer_model = load_model(config['train']['saved_weights_name'])
    infer_model.load_weights(config['train']['saved_weights_name'])
    infer_model.save(config['train']['saved_weights_name'])
    ###############################