浏览代码

Adding training configs

/hw20-segmentation
Jonathan Harper 5 年前
当前提交
4e7a1170
共有 2 个文件被更改,包括 54 次插入0 次删除
  1. 26
      config/ppo/VisualFoodCollector.yaml
  2. 28
      config/sac/VisualFoodCollector.yaml

26
config/ppo/VisualFoodCollector.yaml


behaviors:
VisualFoodCollector:
trainer_type: ppo
hyperparameters:
batch_size: 1024
buffer_size: 10240
learning_rate: 0.0003
beta: 0.005
epsilon: 0.2
lambd: 0.95
num_epoch: 3
learning_rate_schedule: constant
network_settings:
normalize: false
hidden_units: 128
num_layers: 2
vis_encode_type: simple
reward_signals:
extrinsic:
gamma: 0.99
strength: 1.0
keep_checkpoints: 5
max_steps: 2000000
time_horizon: 64
summary_freq: 10000
threaded: true

28
config/sac/VisualFoodCollector.yaml


behaviors:
VisualFoodCollector:
trainer_type: sac
hyperparameters:
learning_rate: 0.0003
learning_rate_schedule: constant
batch_size: 256
buffer_size: 100000
buffer_init_steps: 20000
tau: 0.005
steps_per_update: 10.0
save_replay_buffer: false
init_entcoef: 0.05
reward_signal_steps_per_update: 10.0
network_settings:
normalize: false
hidden_units: 128
num_layers: 2
vis_encode_type: nature_cnn
reward_signals:
extrinsic:
gamma: 0.99
strength: 1.0
keep_checkpoints: 5
max_steps: 2000000
time_horizon: 64
summary_freq: 10000
threaded: true
正在加载...
取消
保存