|
|
|
|
|
|
|
|
|
|
init_venv() |
|
|
|
|
|
|
|
# Copy the default training config but override the max_steps parameter |
|
|
|
override_config_file("config/trainer_config.yaml", "override.yaml", max_steps=100) |
|
|
|
# Copy the default training config but override the max_steps parameter, |
|
|
|
# and reduce the batch_size and buffer_size enough to ensure an update step happens. |
|
|
|
override_config_file( |
|
|
|
"config/trainer_config.yaml", |
|
|
|
"override.yaml", |
|
|
|
max_steps=100, |
|
|
|
batch_size=10, |
|
|
|
buffer_size=10, |
|
|
|
) |
|
|
|
|
|
|
|
# TODO pass scene name and exe destination to build |
|
|
|
# TODO make sure we fail if the exe isn't found - see MLA-559 |
|
|
|