use normalizing on broadcast tiles

2018-06-24 12:18:49 +02:00 · 2018-06-24 12:18:49 +02:00 · 08148c6736
parent 18e4376aae
commit 08148c6736
1 changed files with 1 additions and 1 deletions
--- a/main.lua
+++ b/main.lua
@ -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