ForeverOMC
commented
13 hours ago Hi, I have the same problem as you. I found that this was due to a segment error caused by the set_freest_gpu function. I tried to modify the function to manually set up the gpu and it worked.
Your work is excellent, and I am very interested in it. However, when reproducing it, it has been stuck at 0 iterations and cannot proceed further. Do you know the reason? thank you
(SDS) b101-8-3090@8-3090:~/xueshuoopt/SDS/SDS$ python sds.py task=[bound,hop,pace,trot] [2024-11-22 17:12:04,055][root][INFO] - Workspace: /home/b101-8-3090/xueshuoopt/SDS/SDS/outputs/sds/2024-11-22_17-12-04 [2024-11-22 17:12:04,055][root][INFO] - Project Root: /home/b101-8-3090/xueshuoopt/SDS/SDS [2024-11-22 17:12:04,055][root][INFO] - Running for 5 iterations [2024-11-22 17:12:04,055][root][INFO] - Training each RF for: 1000 iterations [2024-11-22 17:12:04,055][root][INFO] - Generating 8 reward function samples per iteration [2024-11-22 17:12:04,055][root][INFO] - Using LLM: gpt-4o [2024-11-22 17:12:04,055][root][INFO] - Imitation Task: Quadruped Robot Trotting [2024-11-22 17:12:04,055][root][INFO] - Demonstration Video: trot.mp4, Crop Option: None [2024-11-22 17:12:16,769][agents][INFO] - Activated Agent SUSGenerator [2024-11-22 17:12:16,769][agents][INFO] - Activated Agent TaskDescriptor [2024-11-22 17:12:39,012][agents][INFO] - Activated Agent ContactSequenceAnalyser [2024-11-22 17:13:21,584][agents][INFO] - Activated Agent GaitAnalyser [2024-11-22 17:13:38,650][agents][INFO] - Activated Agent TaskRequirementAnalyser [2024-11-22 17:13:58,834][root][INFO] - Iteration 0: Generating 8 samples with gpt-4o [2024-11-22 17:14:58,631][root][INFO] - Iteration 0: Prompt Tokens: 3117, Completion Tokens: 5128, Total Tokens: 11362 [2024-11-22 17:14:58,631][root][INFO] - Iteration 0: Processing Code Run 0