Skip to content

Commit

Permalink
updated readme with compability of releases
Browse files Browse the repository at this point in the history
  • Loading branch information
riannevdberg committed Jul 5, 2019
1 parent 4fbf408 commit 32dde9b
Showing 1 changed file with 14 additions and 2 deletions.
16 changes: 14 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -8,11 +8,23 @@ Rianne van den Berg*, Leonard Hasenclever*, Jakub Tomczak, Max Welling
*Equal contribution

## Requirements
The code is compatible with:
The latest release of the code is compatible with:

* `pytorch 1.0.0`

* `python 3.7`

Thanks to Martin Engelcke for adapting the code to provide this compatibility.
<br/>

Version v0.3.0_2.7 is compatible with:

* `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 vers\
ion.

* `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
The experiments can be run on the following datasets:
* static MNIST: dataset is in data folder;
Expand Down

0 comments on commit 32dde9b

Please sign in to comment.