added comment on BCELoss for later pytorch versions in README

riannevdberg · Nov 5, 2018 · 87aa411 · 87aa411
1 parent f94e72f
commit 87aa411
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -10,7 +10,7 @@ Rianne van den Berg*, Leonard Hasenclever*, Jakub Tomczak, Max Welling
 ## Requirements
 The code is compatible with:
 
-  * `pytorch 0.3.0`
+  * `pytorch 0.3.0` **WARNING**: More recent versions of pytorch have different default flags for the binary cross entropy loss module: nn.BCELoss(). You have to adapt the appropriate flags if you want to port this code to a later version.  
   * `python 2.7`
 
 ## Data