From c9371bc72b9afd93f82769e519b613808eea9769 Mon Sep 17 00:00:00 2001 From: Daniel Saavedra Date: Tue, 23 Aug 2022 20:46:36 -0400 Subject: [PATCH] Update train.py --- keras-yolo3-master/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/keras-yolo3-master/train.py b/keras-yolo3-master/train.py index dac410f..adc2242 100644 --- 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']) ###############################