train from scratch

cc227f26 · dzorlu · 8bb17216 · cc227f26
Commit cc227f26 authored Oct 01, 2020 by dzorlu
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

train.py train.py +1 -1

No files found.
--- a/train.py
+++ b/train.py
@@ -228,7 +228,7 @@ def main():
    logger.info("creating environment")
    environment = make_environment(num_actions=NUMBER_OF_DISCRETE_ACTIONS,
                                   k_means_path=model_dir,
-                                   train=False,
+                                   train=True,
                                   dat_loader=data)
    spec = specs.make_environment_spec(environment)