Skip to content

Commit

Permalink
try alternate yml from dima-babyai-v1.1
Browse files Browse the repository at this point in the history
  • Loading branch information
dyth committed Jun 29, 2020
1 parent 3867dea commit 4f4e8e8
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions .travis.yml
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,12 @@ before_install:
# command to install dependencies
install:
- pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp35-cp35m-linux_x86_64.whl
- pip3 install flake8
- pip3 install scikit-build
- pip3
- pip3 install --editable .
- pip3 install flake8
# - pip3 install flake8
# - pip3 install scikit-build
# - pip3 install --editable .

# command to run tests
script:
Expand All @@ -22,7 +25,7 @@ script:
- ./run_tests.py

# Quickly exercise the RL training code
- time python3 -m scripts.train_rl --env BabyAI-GoToObj-v0 --algo ppo --procs 4 --batch-size 80 --log-interval 1 --save-interval 2 --val-episodes 10 --frames 300 --arch cnn1 --instr-dim 16 --image-dim 16 --memory-dim 16
- time python3 -m scripts.train_rl --env BabyAI-GoToObj-v0 --algo ppo --procs 4 --batch-size 80 --log-interval 1 --save-interval 2 --val-episodes 10 --frames 300 --arch original --instr-dim 16 --image-dim 16 --memory-dim 16

# Check that the bot works on a few episodes of Boss Level
- python3 -m scripts.eval_bot --level BossLevel --num_runs 50
Expand Down

0 comments on commit 4f4e8e8

Please sign in to comment.