diff --git a/config.lua b/config.lua index 3106c45..5fa567b 100644 --- a/config.lua +++ b/config.lua @@ -50,19 +50,27 @@ local common_cfg = { } local cfg = { - -- best performance so far, albeit a bit slow to train. - -- NOTE: uncomment for original values used. - log_fn = 'log.csv', -- can be nil to disable logging. - params_fn = nil, -- can be nil to generate based on param count. + log_fn = 'logs-xnes.csv', + params_fn = 'params-xnes.txt', + + decrement_reward = true, + score_multiplier = 5, + + starting_world = 1, + starting_level = 1, starting_lives = 1, + cap_time = 300, + deterministic = true, - layernorm = true, - epoch_trials = 24, --20, - epoch_top_trials = 18, --10, - ars_lips = true, - deviation = 0.05, --0.1, - weight_decay = 0.0002, --0.0004, - learning_rate = 2.0, --1.0 + + epoch_trials = 50, + epoch_top_trials = 9999, + negate_trials = false, + + es = 'xnes', + learning_rate = 0.14, + deviation = 1.0, + weight_decay = 0.0, } -- TODO: so, uhh..