javier/Photovoltaic_Fault_Detector
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

solved weight problem

Browse Source
This commit is contained in:
Daniel Saavedra
2022-08-25 00:52:15 -04:00
committed by GitHub
parent c9371bc72b
commit 3f160ae076
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
keras-yolo3-master/train.py
View File

@@ -272,11 +272,7 @@ def _main_(args):
callbacks = callbacks callbacks = callbacks
) )
train_model.load_weights(config['train']['saved_weights_name'])
# 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'])
infer_model.save(config['train']['saved_weights_name']) infer_model.save(config['train']['saved_weights_name'])
############################### ###############################
# Run the evaluation # Run the evaluation