Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 7 additions & 2 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -56,5 +56,10 @@ htmlcov/

**/wandb/
runs/
kens_ckpts/
test.ipynb
ckpt_data/
test.ipynb

**/LCRL/
**/LCRL_SAC/

**/local_config.py
1 change: 0 additions & 1 deletion examples/baselines/dreamerv3-torch/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -203,7 +203,6 @@ class Args:
hybrid_steps: int = 1_000_000
hybrid: bool = True


use_gp: bool = True
filter_thresh: float = 0.2
num_runs: int = 1
Expand Down
4 changes: 2 additions & 2 deletions examples/baselines/dreamerv3-torch/dreamer.py
Original file line number Diff line number Diff line change
Expand Up @@ -190,7 +190,8 @@ class Args:

gamma_lx: float = 0.75
# offline_data_path: str = '/home/kensuke/ManiSkill/examples/baselines/ppo/runs/BlockTopple-v0__ppo_rgb__1__1753308792/test_videos/trajectory.rgb.pd_ee_delta_pose.physx_cuda.h5'
offline_data_path: str = '/home/clown2/Desktop/Work/Research/ManiSkill/examples/baselines/ppo/runs/BlockTopple-v0__ppo_rgb__1__1753474701/test_videos/trajectory.rgb.pd_ee_delta_pose.physx_cuda.h5'
# offline_data_path: str = '/home/clown2/Desktop/Work/Research/ManiSkill/Maniskill/examples/baselines/ppo/runs/BlockTopple-v0__ppo_rgb__1__1756574937/test_videos/trajectory.rgb.pd_ee_delta_pose.physx_cuda.h5'
offline_data_path: str = '/home/clown2/Desktop/Work/Research/ManiSkill/Maniskill/examples/baselines/dreamerv3-torch/runs/FilterRolloutGP_20250831-214144/videos/trajectory_mixed.h5'
pretrain: int = 500
hybrid_steps: int = 300_000
hybrid: bool = True
Expand Down Expand Up @@ -237,7 +238,6 @@ def combine_dictionaries(
return combined



class Dreamer(nn.Module):
def __init__(self, obs_space, act_space, args, logger, dataset, expert_dataset=None):
super(Dreamer, self).__init__()
Expand Down
3 changes: 0 additions & 3 deletions examples/baselines/dreamerv3-torch/dreamer_rollout_filter.py
Original file line number Diff line number Diff line change
Expand Up @@ -623,9 +623,6 @@ def main(args):
else:
run_name = 'FilterRolloutNoGP'

# args.logdir = f"runs/{run_name}"
# add time to run name

run_name = f"{run_name}_{time.strftime('%Y%m%d-%H%M%S')}"

args.logdir = f"runs/{run_name}"
Expand Down
Loading