From 08148c67365a4a64740f3ae35b2d2c75bff08ab4 Mon Sep 17 00:00:00 2001 From: Connor Olding Date: Sun, 24 Jun 2018 12:18:49 +0200 Subject: [PATCH] use normalizing on broadcast tiles --- main.lua | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.lua b/main.lua index 5eb3d24..a7eb30f 100644 --- 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