use experimental config/network
This commit is contained in:
parent
401effbc23
commit
19cd10382f
2 changed files with 18 additions and 14 deletions
23
config.lua
23
config.lua
|
@ -53,27 +53,28 @@ local common_cfg = {
|
||||||
}
|
}
|
||||||
|
|
||||||
local cfg = {
|
local cfg = {
|
||||||
log_fn = 'logs-xnes.csv',
|
log_fn = 'logs-snes.csv',
|
||||||
params_fn = 'params-xnes.txt',
|
params_fn = 'params-snes.txt',
|
||||||
|
|
||||||
decrement_reward = true,
|
decrement_reward = true,
|
||||||
score_multiplier = 5,
|
score_multiplier = 5,
|
||||||
|
|
||||||
starting_world = 1,
|
starting_world = 0,
|
||||||
starting_level = 1,
|
starting_level = 1,
|
||||||
starting_lives = 1,
|
--starting_lives = 1,
|
||||||
cap_time = 300,
|
cap_time = 300,
|
||||||
|
|
||||||
deterministic = true,
|
deterministic = false, --true,
|
||||||
|
|
||||||
epoch_trials = 50,
|
epoch_trials = 32,
|
||||||
epoch_top_trials = 9999,
|
epoch_top_trials = 9999,
|
||||||
negate_trials = false,
|
negate_trials = true,
|
||||||
|
|
||||||
es = 'xnes',
|
es = 'snes',
|
||||||
learning_rate = 0.14,
|
learning_rate = 0.5,
|
||||||
deviation = 1.0,
|
mean_adapt = 0.5,
|
||||||
weight_decay = 0.0,
|
deviation = 0.5,
|
||||||
|
weight_decay = 0.025,
|
||||||
}
|
}
|
||||||
|
|
||||||
-- TODO: so, uhh..
|
-- TODO: so, uhh..
|
||||||
|
|
9
main.lua
9
main.lua
|
@ -148,10 +148,13 @@ local function make_network(input_size)
|
||||||
nn_x = nn.Input({input_size})
|
nn_x = nn.Input({input_size})
|
||||||
nn_tx = nn.Input({gcfg.tile_count})
|
nn_tx = nn.Input({gcfg.tile_count})
|
||||||
nn_ty = nn_tx:feed(nn.Embed(#game.valid_tiles, 2))
|
nn_ty = nn_tx:feed(nn.Embed(#game.valid_tiles, 2))
|
||||||
nn_tz = nn_ty:feed(nn.Reshape{13, 17 * 2})
|
|
||||||
nn_tz = nn_tz:feed(nn.DenseBroadcast(5))
|
nn_tz = nn_ty
|
||||||
nn_tz = nn_tz:feed(nn.Relu())
|
--nn_tz = nn_tz:feed(nn.Reshape{13, 17 * 2})
|
||||||
|
--nn_tz = nn_tz:feed(nn.DenseBroadcast(5))
|
||||||
|
--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.
|
||||||
|
|
||||||
nn_y = nn.Merge()
|
nn_y = nn.Merge()
|
||||||
nn_x:feed(nn_y)
|
nn_x:feed(nn_y)
|
||||||
nn_tz:feed(nn_y)
|
nn_tz:feed(nn_y)
|
||||||
|
|
Loading…
Add table
Reference in a new issue