Merge pull request #13 from ethancaballero/patch-3

Install recent pytorch version to reduce leak
stan-stripe · Mar 27, 2017 · e19ac39 · e19ac39
2 parents 85c7efd + 9ba6b4b
commit e19ac39
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -16,6 +16,12 @@ OMP_NUM_THREADS=1 python main.py --env-name "PongDeterministic-v3" --num-process
 
 This code runs evaluation in a separate thread in addition to 16 processes.
 
+Note:
+Install most recent nightly build (version '0.1.10+2fd4d08' or later) of PyTorch via this command to prevent memory leaks:
+`
+pip install git+https://github.com/pytorch/pytorch
+`
+
 ## Results
 
 With 16 processes it converges for PongDeterministic-v3 in 15 minutes.