Skip to content

Commit

Permalink
add BC FoodCollector config (#2987)
Browse files Browse the repository at this point in the history
  • Loading branch information
Chris Elion authored Nov 27, 2019
1 parent 5dc92f1 commit 6063f7e
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions config/offline_bc_config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,20 @@ default:
memory_size: 256
demo_path: ./UnitySDK/Assets/Demonstrations/<Your_Demo_File>.demo

FoodCollector:
trainer: offline_bc
batch_size: 64
summary_freq: 1000
max_steps: 5.0e4
batches_per_epoch: 10
use_recurrent: false
hidden_units: 128
learning_rate: 3.0e-4
num_layers: 2
sequence_length: 32
memory_size: 256
demo_path: ./demos/ExpertFood.demo

Hallway:
trainer: offline_bc
max_steps: 5.0e5
Expand All @@ -25,3 +39,4 @@ Hallway:
memory_size: 256
sequence_length: 32
demo_path: ./demos/ExpertHallway.demo

0 comments on commit 6063f7e

Please sign in to comment.