Fix segmentation fault during destruction

2024-11-04 23:15:18 +08:00
parent 84bfe2e761
commit 361e57fc2b
6 changed files with 24 additions and 18 deletions
--- a/run/run_yopo.py
+++ b/run/run_yopo.py
@@ -113,6 +113,7 @@ def main():
        model.policy.load_state_dict(saved_variables["state_dict"], strict=False)
        model.test_policy(num_rollouts=20)

+    print("Run YOPO Finish!")

 if __name__ == "__main__":
    main()