models.py clean up

eriklindernoren · Apr 12, 2019 · fb9a558 · fb9a558
1 parent 15e1b7e
commit fb9a558
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -4,9 +4,7 @@
     <img src="assets/zurich.jpg" width="900"\>
 </p>
 
-PyTorch implementation of Fast Neural Style Transfer.
-
-[[Paper](https://cs.stanford.edu/people/jcjohns/eccv16/)] [[Authors' Lua Implementation](https://github.com/jcjohnson/fast-neural-style)]
+PyTorch implementation of [Fast Neural Style Transfer](https://cs.stanford.edu/people/jcjohns/eccv16/) ([official Lua implementation](https://github.com/jcjohnson/fast-neural-style)).
 
 I have made some trained models available [here](https://drive.google.com/drive/folders/1aRD6zakhcDImN2Y54qAT6f4801iLcCLB?usp=sharing).
 

diff --git a/models.py b/models.py
@@ -88,6 +88,6 @@ def forward(self, x):
         x = self.block(x)
         if self.norm is not None:
             x = self.norm(x)
-        if self.relu is not None:
+        if self.relu:
             x = F.relu(x)
         return x