浏览代码

larger feature size

/develop/bisim-sac-transfer
yanchaosun 5 年前
当前提交
716336bf
共有 3 个文件被更改,包括 7 次插入7 次删除
  1. 6
      config/sac_transfer/Reacher.yaml
  2. 6
      config/sac_transfer/ReacherTransfer.yaml
  3. 2
      ml-agents/mlagents/trainers/sac_transfer/optimizer.py

6
config/sac_transfer/Reacher.yaml


reward_signal_steps_per_update: 20.0
encoder_layers: 1
policy_layers: 2
forward_layers: 2
forward_layers: 0
feature_size: 64
action_feature_size: 16
feature_size: 128
action_feature_size: 32
separate_policy_train: true
separate_policy_net: true
separate_model_train: true

6
config/sac_transfer/ReacherTransfer.yaml


reward_signal_steps_per_update: 20.0
encoder_layers: 1
policy_layers: 2
forward_layers: 2
forward_layers: 0
feature_size: 64
action_feature_size: 16
feature_size: 128
action_feature_size: 32
separate_policy_train: true
separate_policy_net: true
reuse_encoder: false

2
ml-agents/mlagents/trainers/sac_transfer/optimizer.py


)
self.model_learning_rate = ModelUtils.create_schedule(
hyperparameters.model_schedule,
lr,
lr * 10,
self.policy.global_step,
int(max_step),
min_value=1e-10,

正在加载...
取消
保存