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

Update train.py

Browse Source
This commit is contained in:
Daniel Saavedra
2022-08-23 20:46:36 -04:00
committed by GitHub
parent f106897120
commit c9371bc72b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -274,8 +274,8 @@ def _main_(args):
# make a GPU version of infer_model for evaluation # make a GPU version of infer_model for evaluation
#if multi_gpu > 1: if multi_gpu > 1:
# infer_model = load_model(config['train']['saved_weights_name']) infer_model = load_model(config['train']['saved_weights_name'])
infer_model.load_weights(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'])
############################### ###############################