浏览代码

Reduce memory sizes

/develop/nopreviousactions
Ervin Teng 5 年前
当前提交
9b0b2fed
共有 2 个文件被更改,包括 8 次插入8 次删除
  1. 8
      config/sac_trainer_config.yaml
  2. 8
      config/trainer_config.yaml

8
config/sac_trainer_config.yaml


learning_rate: 3.0e-4
learning_rate_schedule: constant
max_steps: 5.0e5
memory_size: 256
memory_size: 128
normalize: false
num_update: 1
train_interval: 1

sequence_length: 32
num_layers: 2
hidden_units: 128
memory_size: 256
memory_size: 128
init_entcoef: 0.1
max_steps: 1.0e7
summary_freq: 1000

sequence_length: 32
num_layers: 1
hidden_units: 128
memory_size: 256
memory_size: 128
gamma: 0.99
batch_size: 64
max_steps: 1.0e7

sequence_length: 32
num_layers: 1
hidden_units: 128
memory_size: 256
memory_size: 128
gamma: 0.99
buffer_size: 1024
batch_size: 64

8
config/trainer_config.yaml


learning_rate: 3.0e-4
learning_rate_schedule: linear
max_steps: 5.0e5
memory_size: 256
memory_size: 128
normalize: false
num_epoch: 3
num_layers: 2

sequence_length: 64
num_layers: 2
hidden_units: 128
memory_size: 256
memory_size: 128
beta: 1.0e-2
num_epoch: 3
buffer_size: 1024

sequence_length: 64
num_layers: 1
hidden_units: 128
memory_size: 256
memory_size: 128
beta: 1.0e-2
num_epoch: 3
buffer_size: 1024

sequence_length: 32
num_layers: 1
hidden_units: 128
memory_size: 256
memory_size: 128
beta: 1.0e-2
num_epoch: 3
buffer_size: 1024

正在加载...
取消
保存