[BugFix]: make training example gracefully exit (#326)

Author: vmoens

examples/ddpg/ddpg.py

@@ -205,7 +205,7 @@ def main(cfg: "DictConfig"):
     print(f"init seed: {cfg.seed}, final seed: {final_seed}")
 
     trainer.train()
-    return (logger.log_dir, trainer._log_dict, trainer.state_dict())
+    return (logger.log_dir, trainer._log_dict)
 
 
 if __name__ == "__main__":

examples/dqn/dqn.py

@@ -175,7 +175,7 @@ def main(cfg: "DictConfig"):
     print(f"init seed: {cfg.seed}, final seed: {final_seed}")
 
     trainer.train()
-    return (logger.log_dir, trainer._log_dict, trainer.state_dict())
+    return (logger.log_dir, trainer._log_dict)
 
 
 if __name__ == "__main__":

examples/ppo/ppo.py

@@ -177,7 +177,7 @@ def main(cfg: "DictConfig"):
     print(f"init seed: {cfg.seed}, final seed: {final_seed}")
 
     trainer.train()
-    return (logger.log_dir, trainer._log_dict, trainer.state_dict())
+    return (logger.log_dir, trainer._log_dict)
 
 
 if __name__ == "__main__":

examples/redq/redq.py

@@ -205,7 +205,7 @@ def main(cfg: "DictConfig"):
     print(f"init seed: {cfg.seed}, final seed: {final_seed}")
 
     trainer.train()
-    return (logger.log_dir, trainer._log_dict, trainer.state_dict())
+    return (logger.log_dir, trainer._log_dict)
 
 
 if __name__ == "__main__":

examples/sac/sac.py

@@ -201,7 +201,7 @@ def main(cfg: "DictConfig"):
     print(f"init seed: {cfg.seed}, final seed: {final_seed}")
 
     trainer.train()
-    return (logger.log_dir, trainer._log_dict, trainer.state_dict())
+    return (logger.log_dir, trainer._log_dict)
 
 
 if __name__ == "__main__":