use normalizing on broadcast tiles
This commit is contained in:
parent
18e4376aae
commit
08148c6736
2
main.lua
2
main.lua
|
@ -154,7 +154,7 @@ local function make_network(input_size)
|
||||||
nn_tz = nn_ty
|
nn_tz = nn_ty
|
||||||
if cfg.reduce_tiles then
|
if cfg.reduce_tiles then
|
||||||
nn_tz = nn_tz:feed(nn.Reshape{11, 17 * 2})
|
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())
|
nn_tz = nn_tz:feed(nn.Relu())
|
||||||
-- note: due to a quirk in Merge, we don't need to flatten nn_tz.
|
-- note: due to a quirk in Merge, we don't need to flatten nn_tz.
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in New Issue
Block a user