notwa/smbot
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

use normalizing on broadcast tiles

Browse source
This commit is contained in:
Connor Olding 2018-06-24 12:18:49 +02:00
parent 18e4376aae
commit 08148c6736
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
main.lua
View file

@ -154,7 +154,7 @@ local function make_network(input_size)
nn_tz = nn_ty
if cfg.reduce_tiles then
nn_tz = nn_tz:feed(nn.Reshape{11, 17 * 2})
nn_tz = nn_tz:feed(nn.DenseBroadcast(5))
nn_tz = nn_tz:feed(nn.DenseBroadcast(5, true))
nn_tz = nn_tz:feed(nn.Relu())
-- note: due to a quirk in Merge, we don't need to flatten nn_tz.
end