浏览代码

increase bullets to 40

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

4
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, [40], self.n_embd # , concat_self=False
self.transformer = ResidualSelfAttention(self.n_embd, [20])
self.transformer = ResidualSelfAttention(self.n_embd, [40])
# self.transformer = SmallestAttention(x_self_len, entities_sizes, self.h_size, self.h_size)
# self.transformer = SmallestAttention(64, [64], self.h_size, self.h_size)
# self.use_fc = True

正在加载...
取消
保存