From aeebb8c5204222d03f7e45013e1ca66b377a8f7b Mon Sep 17 00:00:00 2001 From: Somshubra Majumdar Date: Thu, 23 Nov 2017 13:57:21 -0600 Subject: [PATCH] Set default weights to None since ImageNet weights arent available yet --- keras_contrib/applications/nasnet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/keras_contrib/applications/nasnet.py b/keras_contrib/applications/nasnet.py index 072172d..34f2e11 100644 --- a/keras_contrib/applications/nasnet.py +++ b/keras_contrib/applications/nasnet.py @@ -262,7 +262,7 @@ def NASNetLarge(input_shape=None, dropout=0.5, use_auxilary_branch=False, include_top=True, - weights='imagenet', + weights=None, input_tensor=None, pooling=None, classes=1000): @@ -333,7 +333,7 @@ def NASNetMobile(input_shape=None, dropout=0.5, use_auxilary_branch=False, include_top=True, - weights='imagenet', + weights=None, input_tensor=None, pooling=None, classes=1000):