url

README.md

@@ -1,6 +1,6 @@
 # MXNet-Gluon-Style-Transfer
 
-This repo provides MXNet Implementation of **[Neural Style Transfer](#neural-style)** and **[MSG-Net](#msg-net)**. We also provide [PyTorch](https://github.com/zhanghang1989/PyTorch-Style-Transfer) and [Torch](https://github.com/zhanghang1989/MSG-Net/) implementations.
+This repo provides MXNet Implementation of **[Neural Style Transfer](#neural-style)** and **[MSG-Net](#real-time-style-transfer)**. We also provide [PyTorch](https://github.com/zhanghang1989/PyTorch-Style-Transfer) and [Torch](https://github.com/zhanghang1989/MSG-Net/) implementations.
 
 **Tabe of content**
 

models/download_model.sh

@@ -1,3 +1,3 @@
 cd models
-wget -O 21styles.params https://www.dropbox.com/s/mchm6tnpsb44zvl/21styles.params?dl=0?dl=1
+wget -O 21styles.params https://apache-mxnet.s3-accelerate.amazonaws.com/gluon/models/21styles-32f7205c5.params
 cd ..

option.py

@@ -3,7 +3,7 @@
 
 class Options():
     def __init__(self):
-        self.parser = argparse.ArgumentParser(description="parser for PyTorch-Style-Transfer")
+        self.parser = argparse.ArgumentParser(description="parser for MXNet-Gluon-Style-Transfer")
         subparsers = self.parser.add_subparsers(title="subcommands", dest="subcommand")
 
         # training args

utils.py

@@ -117,3 +117,5 @@ def init_vgg_params(vgg, model_folder, ctx):
     if not os.path.exists(os.path.join(model_folder, 'mxvgg.params')):
         os.system('wget https://www.dropbox.com/s/7c92s0guekwrwzf/mxvgg.params?dl=1 -O' + os.path.join(model_folder, 'mxvgg.params'))
     vgg.collect_params().load(os.path.join(model_folder, 'mxvgg.params'), ctx=ctx)
+    for param in vgg.collect_params().values():
+        param.grad_req = 'null'