forge/ethyr/torch/policy.py [91:100]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
   def forward(self, conv, flat, ents):
      tiles, nents = conv[0], conv[1]
      tiles = self.embed(tiles.view(-1).long()).view(-1)
      nents = nents.view(-1)
      conv = torch.cat((tiles, nents)) 
      ents = self.ent1(ents)
      ents, _ = torch.max(ents, 0)
      x = torch.cat((conv.view(-1), flat, ents)).view(1, -1)
      x = torch.nn.functional.relu(self.fc1(x))
      return x
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



forge/ethyr/torch/policy.py [150:159]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
   def forward(self, conv, flat, ents):
      tiles, nents = conv[0], conv[1]
      tiles = self.embed(tiles.view(-1).long()).view(-1)
      nents = nents.view(-1)
      conv = torch.cat((tiles, nents)) 
      ents = self.ent1(ents)
      ents, _ = torch.max(ents, 0)
      x = torch.cat((conv.view(-1), flat, ents)).view(1, -1)
      x = torch.nn.functional.relu(self.fc1(x))
      return x
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -