Skip to content

Commit e19ac39

Browse files
authored
Merge pull request #13 from ethancaballero/patch-3
Install recent pytorch version to reduce leak
2 parents 85c7efd + 9ba6b4b commit e19ac39

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

README.md

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,12 @@ OMP_NUM_THREADS=1 python main.py --env-name "PongDeterministic-v3" --num-process
1616

1717
This code runs evaluation in a separate thread in addition to 16 processes.
1818

19+
Note:
20+
Install most recent nightly build (version '0.1.10+2fd4d08' or later) of PyTorch via this command to prevent memory leaks:
21+
`
22+
pip install git+https://github.com/pytorch/pytorch
23+
`
24+
1925
## Results
2026

2127
With 16 processes it converges for PongDeterministic-v3 in 15 minutes.

0 commit comments

Comments
 (0)