[incubator-mxnet] branch master updated: Update fit.py (#10685)

[zh...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new 9dce757  Update fit.py (#10685)
9dce757 is described below

commit 9dce75795577ca2a75d58fabbd7e133ee8e0c668
Author: Arun Das <ar...@my.utsa.edu>
AuthorDate: Wed May 2 23:07:18 2018 -0500

    Update fit.py (#10685)
    
    * Update fit.py
    
    args.network is None during finetuing using fine-tune.py file which is provided. Hence, line 241 throws an error while trying to see if 'vgg' is in args.network. Patching it by checking if args.network is None before seeing if 'vgg' is present in network name.
    
    * Update fit.py
---
 example/image-classification/common/fit.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/example/image-classification/common/fit.py b/example/image-classification/common/fit.py
index 9412b6f..8189a0a 100755
--- a/example/image-classification/common/fit.py
+++ b/example/image-classification/common/fit.py
@@ -238,7 +238,7 @@ def fit(args, network, data_loader, **kwargs):
             # AlexNet will not converge using Xavier
             initializer = mx.init.Normal()
             # VGG will not trend to converge using Xavier-Gaussian
-        elif 'vgg' in args.network:
+        elif args.network and 'vgg' in args.network:
             initializer = mx.init.Xavier()
         else:
             initializer = mx.init.Xavier(

-- 
To stop receiving notification emails like this one, please contact
zhasheng@apache.org.