Adam/Dribble
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

gather expert data

Browse Source
This commit is contained in:
MagDish 2024-09-09 17:35:21 +08:00
parent c6c4411beb
commit 43f905e634
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
behaviors/custom/Dribble/Dribble.py
View File

@ -208,7 +208,8 @@ class Dribble():
if self.phase >= WIND_DOWN_STEPS - 5: if self.phase >= WIND_DOWN_STEPS - 5:
self.phase = 0 self.phase = 0
return True return True
np.array(self.gather_expert_data.append((self.obs, self.act)), dtype=object) if self.obs != None and self.act != None:
np.array(self.gather_expert_data.append((self.obs, self.act)), dtype=object)
size = len(self.gather_expert_data) size = len(self.gather_expert_data)
if size > 5e4: if size > 5e4:
np.save('/home/apollo3d/code/TEST_Dribble/behaviors/custom/Dribble/expert_data.npy', self.gather_expert_data) np.save('/home/apollo3d/code/TEST_Dribble/behaviors/custom/Dribble/expert_data.npy', self.gather_expert_data)