diff --git a/README.md b/README.md index 8de8748..c2853a6 100644 --- 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