From 3f160ae07654992d7abfb605098a54352ed72df9 Mon Sep 17 00:00:00 2001 From: Daniel Saavedra Date: Thu, 25 Aug 2022 00:52:15 -0400 Subject: [PATCH] solved weight problem --- keras-yolo3-master/train.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/keras-yolo3-master/train.py b/keras-yolo3-master/train.py index adc2242..cc14742 100644 --- a/keras-yolo3-master/train.py +++ b/keras-yolo3-master/train.py @@ -272,11 +272,7 @@ def _main_(args): callbacks = callbacks ) - - # make a GPU version of infer_model for evaluation - if multi_gpu > 1: - infer_model = load_model(config['train']['saved_weights_name']) - infer_model.load_weights(config['train']['saved_weights_name']) + train_model.load_weights(config['train']['saved_weights_name']) infer_model.save(config['train']['saved_weights_name']) ############################### # Run the evaluation