Skip to content

Commit

Permalink
Install recent pytorch version to reduce leak
Browse files Browse the repository at this point in the history
  • Loading branch information
ethancaballero authored Mar 27, 2017
1 parent 85c7efd commit 9ba6b4b
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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.
Expand Down

0 comments on commit 9ba6b4b

Please sign in to comment.