From caab998113f102b39cf2b1caccb2af89947ab95a Mon Sep 17 00:00:00 2001 From: Chris Elion Date: Wed, 27 Nov 2019 10:10:54 -0800 Subject: [PATCH] add BC FoodCollector config --- config/offline_bc_config.yaml | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/config/offline_bc_config.yaml b/config/offline_bc_config.yaml index 5c0fd4b6c0..7d071505c5 100644 --- a/config/offline_bc_config.yaml +++ b/config/offline_bc_config.yaml @@ -12,6 +12,20 @@ default: memory_size: 256 demo_path: ./UnitySDK/Assets/Demonstrations/.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 @@ -25,3 +39,4 @@ Hallway: memory_size: 256 sequence_length: 32 demo_path: ./demos/ExpertHallway.demo +