浏览代码

use target

/develop/bisim-sac-transfer
yanchaosun 4 年前
当前提交
8c03c82a
共有 4 个文件被更改,包括 5 次插入5 次删除
  1. 2
      config/sac_transfer/3DBall.yaml
  2. 2
      config/sac_transfer/3DBallHard.yaml
  3. 4
      config/sac_transfer/3DBallHardTransfer.yaml
  4. 2
      ml-agents/mlagents/trainers/sac_transfer/network.py

2
config/sac_transfer/3DBall.yaml


separate_policy_train: true
# separate_value_train: true
separate_model_train: true
reuse_encoder: true
reuse_encoder: false
in_epoch_alter: false
in_batch_alter: true
use_op_buffer: false

2
config/sac_transfer/3DBallHard.yaml


separate_policy_train: true
# separate_value_train: true
separate_model_train: true
reuse_encoder: true
reuse_encoder: false
in_epoch_alter: false
in_batch_alter: true
use_op_buffer: false

4
config/sac_transfer/3DBallHardTransfer.yaml


# separate_value_net: true
separate_policy_train: true
# separate_value_train: true
reuse_encoder: true
reuse_encoder: false
in_epoch_alter: false
in_batch_alter: false
use_op_buffer: false

use_transfer: true
load_model: true
train_model: false
transfer_path: "results/ball/3DBall"
transfer_path: "results/ball-targ/3DBall"
network_settings:
normalize: true
hidden_units: 64

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


self.processed_vector_in,
vis_encode_type,
encoder_layers=encoder_layers,
scope="encoding",
scope="target_enc",
reuse=True
)
if separate_train:

正在加载...
取消
保存