dodox/dppo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

squash commits

Browse Source
This commit is contained in:
allenzren
2024-09-11 21:09:17 -04:00
parent 8ce0aa1485
commit 2ddf63b8f5
200 changed files with 1240 additions and 1186 deletions
Download Patch File Download Diff File Expand all files Collapse all files
script/dataset/filter_d3il_avoid_data.py
+1 -1
View File
@@ -245,7 +245,7 @@ def make_dataset(
plot(out_val, name="val-trajs.png")
# Save to np file
save_train_path = os.path.join(save_dir, save_name_prefix + "train.pkl")
save_train_path = os.path.join(save_dir, save_name_prefix + "train.npz")
save_val_path = os.path.join(save_dir, save_name_prefix + "val.pkl")
with open(save_train_path, "wb") as f:
pickle.dump(out_train, f)
script/dataset/get_d4rl_dataset.py
+1 -1
View File
@@ -142,7 +142,7 @@ def make_dataset(env_name, save_dir, save_name_prefix, val_split, logger):
prev_index = cur_index + 1
# Save to np file
save_train_path = os.path.join(save_dir, save_name_prefix + "train.pkl")
save_train_path = os.path.join(save_dir, save_name_prefix + "train.npz")
save_val_path = os.path.join(save_dir, save_name_prefix + "val.pkl")
with open(save_train_path, "wb") as f:
pickle.dump(out_train, f)
script/dataset/process_d3il_dataset.py
+1 -1
View File
@@ -192,7 +192,7 @@ def make_dataset(load_path, save_dir, save_name_prefix, env_type, val_split):
plot(out_val, name="val-trajs.png")
# Save to np file
save_train_path = os.path.join(save_dir, save_name_prefix + "train.pkl")
save_train_path = os.path.join(save_dir, save_name_prefix + "train.npz")
save_val_path = os.path.join(save_dir, save_name_prefix + "val.pkl")
with open(save_train_path, "wb") as f:
pickle.dump(out_train, f)
script/dataset/process_robomimic_dataset.py
+1 -1
View File
@@ -303,7 +303,7 @@ def make_dataset(
val_episode_reward_all.append(np.sum(data_traj["rewards"]))
# Save to np file
save_train_path = os.path.join(save_dir, save_name_prefix + "train.pkl")
save_train_path = os.path.join(save_dir, save_name_prefix + "train.npz")
save_val_path = os.path.join(save_dir, save_name_prefix + "val.pkl")
with open(save_train_path, "wb") as f:
pickle.dump(out_train, f)