Update core.py

aibharata · Mar 27, 2020 · 89acdd7 · 89acdd7
1 parent 2fb71bb
commit 89acdd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/easyai/chief/core.py b/easyai/chief/core.py
@@ -62,7 +62,7 @@ def load_model_and_weights(modelName, summary = False):
 	model.load_weights(modelName+'_wgts.h5')
 	return model
 
-def modelManager(modelName,x_train, OUTPUT_CLASSES, RETRAIN_MODEL, AI_NAME= 'tinyMedNet'):
+def modelManager(modelName,x_train, OUTPUT_CLASSES, RETRAIN_MODEL, AI_NAME= 'tinyNet'):
 	if RETRAIN_MODEL== True:
 		if check_model_exists(modelName):
 			model = load_model_and_weights(modelName = modelName)