train from scratch

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)