浏览代码

Update visual hyperparameters (#1118)

/develop-generalizationTraining-TrainerController
GitHub 6 年前
当前提交
d0158b01
共有 1 个文件被更改,包括 36 次插入3 次删除
  1. 39
      python/trainer_config.yaml

39
python/trainer_config.yaml


batch_size: 1024
beta: 5.0e-3
buffer_size: 10240
max_steps: 1.0e5
BouncerBrain:
normalize: true

VisualPyramidBrain:
use_curiosity: true
curiosity_strength: 0.01
curiosity_enc_size: 256
batch_size: 32
buffer_size: 1024
batch_size: 64
buffer_size: 2024
num_layers: 2
num_layers: 1
beta: 1.0e-2
max_steps: 5.0e5
num_epoch: 3

num_epoch: 3
buffer_size: 1024
batch_size: 128
max_steps: 5.0e5
summary_freq: 1000
time_horizon: 64
VisualHallwayBrain:
use_recurrent: true
sequence_length: 64
num_layers: 1
hidden_units: 128
memory_size: 256
beta: 1.0e-2
gamma: 0.99
num_epoch: 3
buffer_size: 1024
batch_size: 64
max_steps: 5.0e5
summary_freq: 1000
time_horizon: 64
VisualPushBlockBrain:
use_recurrent: true
sequence_length: 32
num_layers: 1
hidden_units: 128
memory_size: 256
beta: 1.0e-2
gamma: 0.99
num_epoch: 3
buffer_size: 1024
batch_size: 64
max_steps: 5.0e5
summary_freq: 1000
time_horizon: 64

正在加载...
取消
保存