Update inception_v4.py

moyanyu · Feb 13, 2017 · 5acfaca · 5acfaca
1 parent f4c8302
commit 5acfaca
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/inception_v4.py b/inception_v4.py
@@ -272,5 +272,5 @@ def inception_v4(num_classes, dropout_keep_prob, weights):
 
     return model
 
-def create_model(num_classes=1001, dropout_keep_prob=0.8, weights=None):
+def create_model(num_classes=1001, dropout_keep_prob=0.2, weights=None):
 	return inception_v4(num_classes, dropout_keep_prob, weights)