remove plotting and remove onnx export

kscalelabs · Dec 2, 2024 · 8bb1f9a · 8bb1f9a
1 parent 9780d6e
commit 8bb1f9a
Showing 1 changed file with 0 additions and 7 deletions.
diff --git a/sim/play.py b/sim/play.py
@@ -83,12 +83,6 @@ def play(args: argparse.Namespace) -> None:
         export_policy_as_jit(ppo_runner.alg.actor_critic, path)
         print("Exported policy as jit script to: ", path)
 
-    # # export policy as a onnx module (used to run it on web)
-    # if args.export_onnx:
-    #     path = ppo_runner.alg.actor_critic
-    #     convert_model_to_onnx(path, ActorCfg(), save_path="policy.onnx")
-    #     print("Exported policy as onnx to: ", path)
-
     # Prepare for logging
     env_logger = Logger(env.dt)
     robot_index = 0
@@ -309,7 +303,6 @@ def play(args: argparse.Namespace) -> None:
                 krec_loggers[env_idx].add_frame(frame)
 
     env_logger.print_rewards()
-    # env_logger.plot_states()
 
     if args.render:
         video.release()