浏览代码

concat self

/layernorm
Andrew Cohen 4 年前
当前提交
90e0e8b2
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents/mlagents/trainers/torch/networks.py

2
ml-agents/mlagents/trainers/torch/networks.py


# self.h_size
# )
self.entity_embedding = EntityEmbeddings(
x_self_len, entities_sizes, [20], self.n_embd, concat_self=False
x_self_len, entities_sizes, [20], self.n_embd # , concat_self=False
)
# self.embedding_norm = torch.nn.LayerNorm(self.n_embd)

正在加载...
取消
保存