Set default weights to None since ImageNet weights arent available yet

2026-06-27 16:10:11 +08:00 · 2017-11-23 13:57:21 -06:00
parent 17ac57d02a
commit aeebb8c520
1 changed files with 2 additions and 2 deletions
@@ -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):