Commit 839daa91 authored by Alexis Brenon's avatar Alexis Brenon
Browse files

馃敡 Use faulty environment en pre-train env

parent 833c14eb
......@@ -27,77 +27,38 @@ local args = {
-- Environments
training_environment = {
class = "smarthome.sweethome.GraphicalAnnotatedSweetHome",
class = "smarthome.sweethome.GraphicalSensorSweetHome",
params = {
map_path = paths.concat("assets", "domus_inferred.svg"),
max_tries = 1,
deterministic = true,
map_path = paths.concat("assets", "domus.svg"),
history_length = 3,
restricted = true,
faulty_sensors = {
'fffecc84d66b', 'fffe725aa973', 'fffe6a74e37b'
},
},
},
testing_environment = {
class = "smarthome.sweethome.GraphicalAnnotatedSweetHome",
class = "smarthome.sweethome.GraphicalSensorSweetHome",
params = {
map_path = paths.concat("assets", "domus_inferred.svg"),
max_tries = 1,
deterministic = true,
map_path = paths.concat("assets", "domus.svg"),
history_length = 3,
restricted = true,
faulty_sensors = {
'fffecc84d66b', 'fffe725aa973', 'fffe6a74e37b'
},
},
},
-- Agent
agent = {
class = "NeuralQLearner",
params = {
preprocess = {
class = "Downsample",
params = {
scale_size = {84,84}
},
},
inference = {
class = "Inference",
params = {
input_size = {1, 84, 84},
conv_layers = {
{
n_filters = 32,
field_size = {width = 8, height = 8},
stride = {width = 4, height = 4},
zero_padding = {width = 0, height = 0}
},
{
n_filters = 64,
field_size = {width = 4, height = 4},
stride = {width = 2, height = 2},
zero_padding = {width = 0, height = 0}
},
{
n_filters = 64,
field_size = {width = 3, height = 3},
stride = {width = 1, height = 1},
zero_padding = {width = 0, height = 0}
}
}
},
},
experience_pool = {
pool_size = 100,
},
learn_start = 50,
target_q = 100,
update_freq = 1,
minibatch_size = 32,
ep_start = 1,
ep_end = 0.5,
ep_endt = 100000,
}
dump = "./outputs/synthetic-raw-agent_400000.t7"
},
-- Experiment
experiment = {
class = "BaseExperiment",
params = {
steps = 300e3,
steps = 400e3,
eval_freq = 5e3,
eval_steps = 5000,
},
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment