Add files using upload-large-folder tool

visual-ogbench/checkpoints/action-prior/cd_noisy/flags.json

@@ -0,0 +1 @@
+{"logtostderr": false, "alsologtostderr": false, "log_dir": "", "v": 0, "verbosity": 0, "logger_levels": {}, "stderrthreshold": "fatal", "showprefixforinfo": true, "run_with_pdb": false, "pdb_post_mortem": false, "pdb": false, "run_with_profiling": false, "profile_file": null, "use_cprofile_for_profiling": true, "only_check_args": false, "chex_n_cpu_devices": 1, "chex_assert_multiple_cpu_devices": false, "test_srcdir": "", "test_tmpdir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_cd_noisy_k5_bs512_s500k_s0_a3_e09_aug05/tmp/absl_testing", "test_random_seed": 301, "test_randomize_ordering_seed": "", "xml_output_file": "", "chex_skip_pmap_variant_if_single_device": true, "run_group": "2026-08-19_yb_GCAWR_cd_noisy_k5_bs512_s500k_s0_a3_e09_aug05", "wandb_mode": "offline", "seed": 0, "env_name": "visual-cube-double-noisy-v0", "dataset_path": null, "validation_fraction": 0.05, "save_dir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_cd_noisy_k5_bs512_s500k_s0_a3_e09_aug05/OGBench/2026-08-19_yb_GCAWR_cd_noisy_k5_bs512_s500k_s0_a3_e09_aug05/sd000_20260819_080318", "restore_path": null, "restore_epoch": null, "eval_only": false, "train_steps": 500000, "log_interval": 5000, "eval_interval": 100000, "save_interval": 100000, "eval_tasks": null, "eval_episodes": 0, "eval_temperature": 0.0, "eval_gaussian": null, "video_episodes": 0, "video_frame_skip": 3, "eval_on_cpu": 1, "agent": {"actor_geom_sample": false, "actor_hidden_dims": [512, 512, 512], "actor_loss": "awr", "actor_p_curgoal": 0.0, "actor_p_randomgoal": 0.0, "actor_p_trajgoal": 1.0, "agent_name": "gciql_chunk", "alpha": 3.0, "batch_size": 512, "chunk_size": 5, "const_std": true, "dataset_class": "GCChunkDataset", "discount": 0.99, "discrete": false, "encoder": "impala_small", "expectile": 0.9, "frame_stack": null, "gc_negative": true, "layer_norm": true, "lr": 0.0003, "p_aug": 0.5, "tau": 0.005, "value_geom_sample": true, "value_hidden_dims": [512, 512, 512], "value_p_curgoal": 0.2, "value_p_randomgoal": 0.3, "value_p_trajgoal": 0.5}, "?": false, "help": false, "helpshort": false, "helpfull": false, "helpxml": false, "only_check_flags": false}

visual-ogbench/checkpoints/action-prior/cd_noisy/params_500000.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0cc1c9b31112b9531777ad396441b63dcb6fe70b56ad91771ff08a0d7c7c8c7e
+size 110803260

visual-ogbench/checkpoints/action-prior/cd_play/flags.json

@@ -0,0 +1 @@
+{"logtostderr": false, "alsologtostderr": false, "log_dir": "", "v": 0, "verbosity": 0, "logger_levels": {}, "stderrthreshold": "fatal", "showprefixforinfo": true, "run_with_pdb": false, "pdb_post_mortem": false, "pdb": false, "run_with_profiling": false, "profile_file": null, "use_cprofile_for_profiling": true, "only_check_args": false, "chex_n_cpu_devices": 1, "chex_assert_multiple_cpu_devices": false, "test_srcdir": "", "test_tmpdir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_cd_play_k5_bs512_s500k_s0_a3_e09_aug05/tmp/absl_testing", "test_random_seed": 301, "test_randomize_ordering_seed": "", "xml_output_file": "", "chex_skip_pmap_variant_if_single_device": true, "run_group": "2026-08-19_yb_GCAWR_cd_play_k5_bs512_s500k_s0_a3_e09_aug05", "wandb_mode": "offline", "seed": 0, "env_name": "visual-cube-double-play-v0", "dataset_path": null, "validation_fraction": 0.05, "save_dir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_cd_play_k5_bs512_s500k_s0_a3_e09_aug05/OGBench/2026-08-19_yb_GCAWR_cd_play_k5_bs512_s500k_s0_a3_e09_aug05/sd000_20260819_080318", "restore_path": null, "restore_epoch": null, "eval_only": false, "train_steps": 500000, "log_interval": 5000, "eval_interval": 100000, "save_interval": 100000, "eval_tasks": null, "eval_episodes": 0, "eval_temperature": 0.0, "eval_gaussian": null, "video_episodes": 0, "video_frame_skip": 3, "eval_on_cpu": 1, "agent": {"actor_geom_sample": false, "actor_hidden_dims": [512, 512, 512], "actor_loss": "awr", "actor_p_curgoal": 0.0, "actor_p_randomgoal": 0.0, "actor_p_trajgoal": 1.0, "agent_name": "gciql_chunk", "alpha": 3.0, "batch_size": 512, "chunk_size": 5, "const_std": true, "dataset_class": "GCChunkDataset", "discount": 0.99, "discrete": false, "encoder": "impala_small", "expectile": 0.9, "frame_stack": null, "gc_negative": true, "layer_norm": true, "lr": 0.0003, "p_aug": 0.5, "tau": 0.005, "value_geom_sample": true, "value_hidden_dims": [512, 512, 512], "value_p_curgoal": 0.2, "value_p_randomgoal": 0.3, "value_p_trajgoal": 0.5}, "?": false, "help": false, "helpshort": false, "helpfull": false, "helpxml": false, "only_check_flags": false}

visual-ogbench/checkpoints/action-prior/cd_play/params_500000.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b71a1f056f6c928a4d8b34ca671393c8615be11044bda80b6fac559ff9d815bb
+size 110803260

visual-ogbench/checkpoints/action-prior/cs_noisy/flags.json

@@ -0,0 +1 @@
+{"logtostderr": false, "alsologtostderr": false, "log_dir": "", "v": 0, "verbosity": 0, "logger_levels": {}, "stderrthreshold": "fatal", "showprefixforinfo": true, "run_with_pdb": false, "pdb_post_mortem": false, "pdb": false, "run_with_profiling": false, "profile_file": null, "use_cprofile_for_profiling": true, "only_check_args": false, "chex_n_cpu_devices": 1, "chex_assert_multiple_cpu_devices": false, "test_srcdir": "", "test_tmpdir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_cs_noisy_k5_bs512_s500k_s0_a3_e09_aug05/tmp/absl_testing", "test_random_seed": 301, "test_randomize_ordering_seed": "", "xml_output_file": "", "chex_skip_pmap_variant_if_single_device": true, "run_group": "2026-08-19_yb_GCAWR_cs_noisy_k5_bs512_s500k_s0_a3_e09_aug05", "wandb_mode": "offline", "seed": 0, "env_name": "visual-cube-single-noisy-v0", "dataset_path": null, "validation_fraction": 0.05, "save_dir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_cs_noisy_k5_bs512_s500k_s0_a3_e09_aug05/OGBench/2026-08-19_yb_GCAWR_cs_noisy_k5_bs512_s500k_s0_a3_e09_aug05/sd000_20260819_080318", "restore_path": null, "restore_epoch": null, "eval_only": false, "train_steps": 500000, "log_interval": 5000, "eval_interval": 100000, "save_interval": 100000, "eval_tasks": null, "eval_episodes": 0, "eval_temperature": 0.0, "eval_gaussian": null, "video_episodes": 0, "video_frame_skip": 3, "eval_on_cpu": 1, "agent": {"actor_geom_sample": false, "actor_hidden_dims": [512, 512, 512], "actor_loss": "awr", "actor_p_curgoal": 0.0, "actor_p_randomgoal": 0.0, "actor_p_trajgoal": 1.0, "agent_name": "gciql_chunk", "alpha": 3.0, "batch_size": 512, "chunk_size": 5, "const_std": true, "dataset_class": "GCChunkDataset", "discount": 0.99, "discrete": false, "encoder": "impala_small", "expectile": 0.9, "frame_stack": null, "gc_negative": true, "layer_norm": true, "lr": 0.0003, "p_aug": 0.5, "tau": 0.005, "value_geom_sample": true, "value_hidden_dims": [512, 512, 512], "value_p_curgoal": 0.2, "value_p_randomgoal": 0.3, "value_p_trajgoal": 0.5}, "?": false, "help": false, "helpshort": false, "helpfull": false, "helpxml": false, "only_check_flags": false}

visual-ogbench/checkpoints/action-prior/cs_noisy/params_500000.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e256272287c4e36a2c2e2bd39c45987bccafa7ef6d8bb72101c6e3244bd93e79
+size 110803260

visual-ogbench/checkpoints/action-prior/cs_play/flags.json

@@ -0,0 +1 @@
+{"logtostderr": false, "alsologtostderr": false, "log_dir": "", "v": 0, "verbosity": 0, "logger_levels": {}, "stderrthreshold": "fatal", "showprefixforinfo": true, "run_with_pdb": false, "pdb_post_mortem": false, "pdb": false, "run_with_profiling": false, "profile_file": null, "use_cprofile_for_profiling": true, "only_check_args": false, "chex_n_cpu_devices": 1, "chex_assert_multiple_cpu_devices": false, "test_srcdir": "", "test_tmpdir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_cs_play_k5_bs512_s500k_s0_a3_e09_aug05/tmp/absl_testing", "test_random_seed": 301, "test_randomize_ordering_seed": "", "xml_output_file": "", "chex_skip_pmap_variant_if_single_device": true, "run_group": "2026-08-19_yb_GCAWR_cs_play_k5_bs512_s500k_s0_a3_e09_aug05", "wandb_mode": "offline", "seed": 0, "env_name": "visual-cube-single-play-v0", "dataset_path": null, "validation_fraction": 0.05, "save_dir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_cs_play_k5_bs512_s500k_s0_a3_e09_aug05/OGBench/2026-08-19_yb_GCAWR_cs_play_k5_bs512_s500k_s0_a3_e09_aug05/sd000_20260819_080318", "restore_path": null, "restore_epoch": null, "eval_only": false, "train_steps": 500000, "log_interval": 5000, "eval_interval": 100000, "save_interval": 100000, "eval_tasks": null, "eval_episodes": 0, "eval_temperature": 0.0, "eval_gaussian": null, "video_episodes": 0, "video_frame_skip": 3, "eval_on_cpu": 1, "agent": {"actor_geom_sample": false, "actor_hidden_dims": [512, 512, 512], "actor_loss": "awr", "actor_p_curgoal": 0.0, "actor_p_randomgoal": 0.0, "actor_p_trajgoal": 1.0, "agent_name": "gciql_chunk", "alpha": 3.0, "batch_size": 512, "chunk_size": 5, "const_std": true, "dataset_class": "GCChunkDataset", "discount": 0.99, "discrete": false, "encoder": "impala_small", "expectile": 0.9, "frame_stack": null, "gc_negative": true, "layer_norm": true, "lr": 0.0003, "p_aug": 0.5, "tau": 0.005, "value_geom_sample": true, "value_hidden_dims": [512, 512, 512], "value_p_curgoal": 0.2, "value_p_randomgoal": 0.3, "value_p_trajgoal": 0.5}, "?": false, "help": false, "helpshort": false, "helpfull": false, "helpxml": false, "only_check_flags": false}

visual-ogbench/checkpoints/action-prior/cs_play/params_500000.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:de4e97b3e2b2bbf562570a00b42194f8a87c56b85c7420ad2391517e52ea8469
+size 110803260

visual-ogbench/checkpoints/action-prior/ct_noisy/flags.json

@@ -0,0 +1 @@
+{"logtostderr": false, "alsologtostderr": false, "log_dir": "", "v": 0, "verbosity": 0, "logger_levels": {}, "stderrthreshold": "fatal", "showprefixforinfo": true, "run_with_pdb": false, "pdb_post_mortem": false, "pdb": false, "run_with_profiling": false, "profile_file": null, "use_cprofile_for_profiling": true, "only_check_args": false, "chex_n_cpu_devices": 1, "chex_assert_multiple_cpu_devices": false, "test_srcdir": "", "test_tmpdir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_ct_noisy_k5_bs512_s500k_s0_a3_e09_aug05/tmp/absl_testing", "test_random_seed": 301, "test_randomize_ordering_seed": "", "xml_output_file": "", "chex_skip_pmap_variant_if_single_device": true, "run_group": "2026-08-19_yb_GCAWR_ct_noisy_k5_bs512_s500k_s0_a3_e09_aug05", "wandb_mode": "offline", "seed": 0, "env_name": "visual-cube-triple-noisy-v0", "dataset_path": null, "validation_fraction": 0.05, "save_dir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_ct_noisy_k5_bs512_s500k_s0_a3_e09_aug05/OGBench/2026-08-19_yb_GCAWR_ct_noisy_k5_bs512_s500k_s0_a3_e09_aug05/sd000_20260819_080318", "restore_path": null, "restore_epoch": null, "eval_only": false, "train_steps": 500000, "log_interval": 5000, "eval_interval": 100000, "save_interval": 100000, "eval_tasks": null, "eval_episodes": 0, "eval_temperature": 0.0, "eval_gaussian": null, "video_episodes": 0, "video_frame_skip": 3, "eval_on_cpu": 1, "agent": {"actor_geom_sample": false, "actor_hidden_dims": [512, 512, 512], "actor_loss": "awr", "actor_p_curgoal": 0.0, "actor_p_randomgoal": 0.0, "actor_p_trajgoal": 1.0, "agent_name": "gciql_chunk", "alpha": 3.0, "batch_size": 512, "chunk_size": 5, "const_std": true, "dataset_class": "GCChunkDataset", "discount": 0.99, "discrete": false, "encoder": "impala_small", "expectile": 0.9, "frame_stack": null, "gc_negative": true, "layer_norm": true, "lr": 0.0003, "p_aug": 0.5, "tau": 0.005, "value_geom_sample": true, "value_hidden_dims": [512, 512, 512], "value_p_curgoal": 0.2, "value_p_randomgoal": 0.3, "value_p_trajgoal": 0.5}, "?": false, "help": false, "helpshort": false, "helpfull": false, "helpxml": false, "only_check_flags": false}

visual-ogbench/checkpoints/action-prior/ct_noisy/params_500000.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:63292b8b146897180dc29d06bf0270ea88cb903c626b75647e957bf5a0f0d19f
+size 110803260

visual-ogbench/checkpoints/action-prior/ct_play/flags.json

@@ -0,0 +1 @@
+{"logtostderr": false, "alsologtostderr": false, "log_dir": "", "v": 0, "verbosity": 0, "logger_levels": {}, "stderrthreshold": "fatal", "showprefixforinfo": true, "run_with_pdb": false, "pdb_post_mortem": false, "pdb": false, "run_with_profiling": false, "profile_file": null, "use_cprofile_for_profiling": true, "only_check_args": false, "chex_n_cpu_devices": 1, "chex_assert_multiple_cpu_devices": false, "test_srcdir": "", "test_tmpdir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_ct_play_k5_bs512_s500k_s0_a3_e09_aug05/tmp/absl_testing", "test_random_seed": 301, "test_randomize_ordering_seed": "", "xml_output_file": "", "chex_skip_pmap_variant_if_single_device": true, "run_group": "2026-08-19_yb_GCAWR_ct_play_k5_bs512_s500k_s0_a3_e09_aug05", "wandb_mode": "offline", "seed": 0, "env_name": "visual-cube-triple-play-v0", "dataset_path": null, "validation_fraction": 0.05, "save_dir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_ct_play_k5_bs512_s500k_s0_a3_e09_aug05/OGBench/2026-08-19_yb_GCAWR_ct_play_k5_bs512_s500k_s0_a3_e09_aug05/sd000_20260819_080318", "restore_path": null, "restore_epoch": null, "eval_only": false, "train_steps": 500000, "log_interval": 5000, "eval_interval": 100000, "save_interval": 100000, "eval_tasks": null, "eval_episodes": 0, "eval_temperature": 0.0, "eval_gaussian": null, "video_episodes": 0, "video_frame_skip": 3, "eval_on_cpu": 1, "agent": {"actor_geom_sample": false, "actor_hidden_dims": [512, 512, 512], "actor_loss": "awr", "actor_p_curgoal": 0.0, "actor_p_randomgoal": 0.0, "actor_p_trajgoal": 1.0, "agent_name": "gciql_chunk", "alpha": 3.0, "batch_size": 512, "chunk_size": 5, "const_std": true, "dataset_class": "GCChunkDataset", "discount": 0.99, "discrete": false, "encoder": "impala_small", "expectile": 0.9, "frame_stack": null, "gc_negative": true, "layer_norm": true, "lr": 0.0003, "p_aug": 0.5, "tau": 0.005, "value_geom_sample": true, "value_hidden_dims": [512, 512, 512], "value_p_curgoal": 0.2, "value_p_randomgoal": 0.3, "value_p_trajgoal": 0.5}, "?": false, "help": false, "helpshort": false, "helpfull": false, "helpxml": false, "only_check_flags": false}

visual-ogbench/checkpoints/action-prior/ct_play/params_500000.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:92f0b0d3c4273add9b658850b42e6985586ff3348b35b5d4a7959fcc1e73ff61
+size 110803260

visual-ogbench/checkpoints/action-prior/scene_noisy/flags.json

@@ -0,0 +1 @@
+{"logtostderr": false, "alsologtostderr": false, "log_dir": "", "v": 0, "verbosity": 0, "logger_levels": {}, "stderrthreshold": "fatal", "showprefixforinfo": true, "run_with_pdb": false, "pdb_post_mortem": false, "pdb": false, "run_with_profiling": false, "profile_file": null, "use_cprofile_for_profiling": true, "only_check_args": false, "chex_n_cpu_devices": 1, "chex_assert_multiple_cpu_devices": false, "test_srcdir": "", "test_tmpdir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_scene_noisy_k5_bs512_s500k_s0_a3_e09_aug05/tmp/absl_testing", "test_random_seed": 301, "test_randomize_ordering_seed": "", "xml_output_file": "", "chex_skip_pmap_variant_if_single_device": true, "run_group": "2026-08-19_yb_GCAWR_scene_noisy_k5_bs512_s500k_s0_a3_e09_aug05", "wandb_mode": "offline", "seed": 0, "env_name": "visual-scene-noisy-v0", "dataset_path": null, "validation_fraction": 0.05, "save_dir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_scene_noisy_k5_bs512_s500k_s0_a3_e09_aug05/OGBench/2026-08-19_yb_GCAWR_scene_noisy_k5_bs512_s500k_s0_a3_e09_aug05/sd000_20260819_080318", "restore_path": null, "restore_epoch": null, "eval_only": false, "train_steps": 500000, "log_interval": 5000, "eval_interval": 100000, "save_interval": 100000, "eval_tasks": null, "eval_episodes": 0, "eval_temperature": 0.0, "eval_gaussian": null, "video_episodes": 0, "video_frame_skip": 3, "eval_on_cpu": 1, "agent": {"actor_geom_sample": false, "actor_hidden_dims": [512, 512, 512], "actor_loss": "awr", "actor_p_curgoal": 0.0, "actor_p_randomgoal": 0.0, "actor_p_trajgoal": 1.0, "agent_name": "gciql_chunk", "alpha": 3.0, "batch_size": 512, "chunk_size": 5, "const_std": true, "dataset_class": "GCChunkDataset", "discount": 0.99, "discrete": false, "encoder": "impala_small", "expectile": 0.9, "frame_stack": null, "gc_negative": true, "layer_norm": true, "lr": 0.0003, "p_aug": 0.5, "tau": 0.005, "value_geom_sample": true, "value_hidden_dims": [512, 512, 512], "value_p_curgoal": 0.2, "value_p_randomgoal": 0.3, "value_p_trajgoal": 0.5}, "?": false, "help": false, "helpshort": false, "helpfull": false, "helpxml": false, "only_check_flags": false}

visual-ogbench/checkpoints/action-prior/scene_noisy/params_500000.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d312013199ff905972158186051ce4d62cca88d50d26b27d284ce2cfa453078
+size 110803260

visual-ogbench/checkpoints/action-prior/scene_play/flags.json

@@ -0,0 +1 @@
+{"logtostderr": false, "alsologtostderr": false, "log_dir": "", "v": 0, "verbosity": 0, "logger_levels": {}, "stderrthreshold": "fatal", "showprefixforinfo": true, "run_with_pdb": false, "pdb_post_mortem": false, "pdb": false, "run_with_profiling": false, "profile_file": null, "use_cprofile_for_profiling": true, "only_check_args": false, "chex_n_cpu_devices": 1, "chex_assert_multiple_cpu_devices": false, "test_srcdir": "", "test_tmpdir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_scene_play_k5_bs512_s500k_s0_a3_e09_aug05/tmp/absl_testing", "test_random_seed": 301, "test_randomize_ordering_seed": "", "xml_output_file": "", "chex_skip_pmap_variant_if_single_device": true, "run_group": "2026-08-19_yb_GCAWR_scene_play_k5_bs512_s500k_s0_a3_e09_aug05", "wandb_mode": "offline", "seed": 0, "env_name": "visual-scene-play-v0", "dataset_path": null, "validation_fraction": 0.05, "save_dir": "/root/data/yyf/ogbench-gciql-chunk-awr-runs/2026-08-19_yb_GCAWR_scene_play_k5_bs512_s500k_s0_a3_e09_aug05/OGBench/2026-08-19_yb_GCAWR_scene_play_k5_bs512_s500k_s0_a3_e09_aug05/sd000_20260819_080318", "restore_path": null, "restore_epoch": null, "eval_only": false, "train_steps": 500000, "log_interval": 5000, "eval_interval": 100000, "save_interval": 100000, "eval_tasks": null, "eval_episodes": 0, "eval_temperature": 0.0, "eval_gaussian": null, "video_episodes": 0, "video_frame_skip": 3, "eval_on_cpu": 1, "agent": {"actor_geom_sample": false, "actor_hidden_dims": [512, 512, 512], "actor_loss": "awr", "actor_p_curgoal": 0.0, "actor_p_randomgoal": 0.0, "actor_p_trajgoal": 1.0, "agent_name": "gciql_chunk", "alpha": 3.0, "batch_size": 512, "chunk_size": 5, "const_std": true, "dataset_class": "GCChunkDataset", "discount": 0.99, "discrete": false, "encoder": "impala_small", "expectile": 0.9, "frame_stack": null, "gc_negative": true, "layer_norm": true, "lr": 0.0003, "p_aug": 0.5, "tau": 0.005, "value_geom_sample": true, "value_hidden_dims": [512, 512, 512], "value_p_curgoal": 0.2, "value_p_randomgoal": 0.3, "value_p_trajgoal": 0.5}, "?": false, "help": false, "helpshort": false, "helpfull": false, "helpxml": false, "only_check_flags": false}

visual-ogbench/checkpoints/action-prior/scene_play/params_500000.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7269e388e7b7a7144ff909530d223f9fc6e2897c51308bba281fe7b134b780a4
+size 110803260

visual-ogbench/checkpoints/latent-path-flow/cd_noisy/checkpoint_200000.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0b81fae7894401d53874f6b2ebd511e6f71a65b0d2c4e26e959d65214ac66577
+size 299898830

visual-ogbench/checkpoints/latent-path-flow/cd_noisy/config.json

@@ -0,0 +1,115 @@
+{
+  "action_block": 5,
+  "architecture": "latent_path_flow_transformer_encoder",
+  "batch_size": 1024,
+  "checkpoint_interval": 25000,
+  "conditioning": "history_goal_time_adaln",
+  "depth": 4,
+  "ema_decay": 0.9999,
+  "embed_dim": 192,
+  "eval_batch_size": 1024,
+  "eval_interval": 10000,
+  "exp_name": "latent_pathflow_ogbench8_cd_noisy_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "ff_dim": 2048,
+  "final_learning_rate": 1e-05,
+  "flow_sampling_steps": 16,
+  "flow_solver": "euler",
+  "goal_sampling": "hiql_uniform_future_same_trajectory",
+  "gradient_clip": 1.0,
+  "hidden_dim": 512,
+  "history_size": 3,
+  "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_cd_noisy__node3_lewm_s3072_e10_z192.h5",
+  "learning_rate": 0.0001,
+  "lewm_checkpoint_sha256": "0f3f5f573f0246b21d0b9866e80df7027a56d6462582f78e426c921042ebc11b",
+  "log_interval": 1000,
+  "loss": "conditional_path_flow_matching_mse",
+  "max_goal_steps": null,
+  "num_episodes": 1000,
+  "num_heads": 8,
+  "num_rows": 1001000,
+  "num_samples": 1,
+  "num_train_episodes": 950,
+  "num_train_transitions": 950000,
+  "num_val_episodes": 50,
+  "num_val_transitions": 50000,
+  "resume": true,
+  "save_dir": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/latent_pathflow_ogbench8_cd_noisy_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "seed": 0,
+  "split_seed": 0,
+  "subgoal_steps": 10,
+  "time_dim": 64,
+  "train_fraction": 0.95,
+  "train_steps": 200000,
+  "validation_manifest": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/manifests/ogbench8_cd_noisy_h50_hist3_k10_n10000.npz",
+  "validation_manifest_metadata": {
+    "action_block": 5,
+    "format": "lewm_latent_subgoal_validation_manifest",
+    "goal_offset": 50,
+    "history_size": 3,
+    "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_cd_noisy__node3_lewm_s3072_e10_z192.h5",
+    "lewm_checkpoint_sha256": "0f3f5f573f0246b21d0b9866e80df7027a56d6462582f78e426c921042ebc11b",
+    "num_episodes": 1000,
+    "num_pairs": 10000,
+    "num_rows": 1001000,
+    "sampling_seed": 1,
+    "split_seed": 0,
+    "subgoal_steps": 10,
+    "train_fraction": 0.95,
+    "validation_episode_ids": [
+      307,
+      613,
+      240,
+      561,
+      570,
+      699,
+      681,
+      936,
+      864,
+      616,
+      317,
+      780,
+      719,
+      753,
+      800,
+      568,
+      302,
+      826,
+      782,
+      73,
+      934,
+      775,
+      889,
+      115,
+      287,
+      191,
+      496,
+      104,
+      176,
+      845,
+      844,
+      7,
+      487,
+      959,
+      351,
+      581,
+      881,
+      639,
+      377,
+      78,
+      632,
+      315,
+      950,
+      29,
+      425,
+      184,
+      504,
+      801,
+      975,
+      607
+    ]
+  },
+  "validation_manifest_sha256": "3df1403f2da5b057b1dc1a96e1fad49bf3faa3a6ef6cd7f0ec0ac8706e5fc403",
+  "validation_pairs": 10000,
+  "warmup_steps": 5000,
+  "weight_decay": 0.0001
+}

visual-ogbench/checkpoints/latent-path-flow/cd_play/checkpoint_200000.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a6118c99afcdb391f50350e2078fe258392290e8d412f58a6f0be653c8b9dc1
+size 299898830

visual-ogbench/checkpoints/latent-path-flow/cd_play/config.json

@@ -0,0 +1,115 @@
+{
+  "action_block": 5,
+  "architecture": "latent_path_flow_transformer_encoder",
+  "batch_size": 1024,
+  "checkpoint_interval": 25000,
+  "conditioning": "history_goal_time_adaln",
+  "depth": 4,
+  "ema_decay": 0.9999,
+  "embed_dim": 192,
+  "eval_batch_size": 1024,
+  "eval_interval": 10000,
+  "exp_name": "latent_pathflow_ogbench8_cd_play_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "ff_dim": 2048,
+  "final_learning_rate": 1e-05,
+  "flow_sampling_steps": 16,
+  "flow_solver": "euler",
+  "goal_sampling": "hiql_uniform_future_same_trajectory",
+  "gradient_clip": 1.0,
+  "hidden_dim": 512,
+  "history_size": 3,
+  "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_cd_play__node3_lewm_s3072_e10_z192.h5",
+  "learning_rate": 0.0001,
+  "lewm_checkpoint_sha256": "ed109fcf37df140987694809c08ab8231819a4e7932497c18c00caaf778fa183",
+  "log_interval": 1000,
+  "loss": "conditional_path_flow_matching_mse",
+  "max_goal_steps": null,
+  "num_episodes": 1000,
+  "num_heads": 8,
+  "num_rows": 1001000,
+  "num_samples": 1,
+  "num_train_episodes": 950,
+  "num_train_transitions": 950000,
+  "num_val_episodes": 50,
+  "num_val_transitions": 50000,
+  "resume": true,
+  "save_dir": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/latent_pathflow_ogbench8_cd_play_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "seed": 0,
+  "split_seed": 0,
+  "subgoal_steps": 10,
+  "time_dim": 64,
+  "train_fraction": 0.95,
+  "train_steps": 200000,
+  "validation_manifest": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/manifests/ogbench8_cd_play_h50_hist3_k10_n10000.npz",
+  "validation_manifest_metadata": {
+    "action_block": 5,
+    "format": "lewm_latent_subgoal_validation_manifest",
+    "goal_offset": 50,
+    "history_size": 3,
+    "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_cd_play__node3_lewm_s3072_e10_z192.h5",
+    "lewm_checkpoint_sha256": "ed109fcf37df140987694809c08ab8231819a4e7932497c18c00caaf778fa183",
+    "num_episodes": 1000,
+    "num_pairs": 10000,
+    "num_rows": 1001000,
+    "sampling_seed": 1,
+    "split_seed": 0,
+    "subgoal_steps": 10,
+    "train_fraction": 0.95,
+    "validation_episode_ids": [
+      307,
+      613,
+      240,
+      561,
+      570,
+      699,
+      681,
+      936,
+      864,
+      616,
+      317,
+      780,
+      719,
+      753,
+      800,
+      568,
+      302,
+      826,
+      782,
+      73,
+      934,
+      775,
+      889,
+      115,
+      287,
+      191,
+      496,
+      104,
+      176,
+      845,
+      844,
+      7,
+      487,
+      959,
+      351,
+      581,
+      881,
+      639,
+      377,
+      78,
+      632,
+      315,
+      950,
+      29,
+      425,
+      184,
+      504,
+      801,
+      975,
+      607
+    ]
+  },
+  "validation_manifest_sha256": "91f710ecba1949c833c1ffce317854c6ffcac2eaac24d339d96ed67143c6b834",
+  "validation_pairs": 10000,
+  "warmup_steps": 5000,
+  "weight_decay": 0.0001
+}

visual-ogbench/checkpoints/latent-path-flow/cs_noisy/checkpoint_200000.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6610c8eea0f0c6edacf73d179f2198713132616e8d6c1403a7cdca71e074272d
+size 299898830

visual-ogbench/checkpoints/latent-path-flow/cs_noisy/config.json

@@ -0,0 +1,115 @@
+{
+  "action_block": 5,
+  "architecture": "latent_path_flow_transformer_encoder",
+  "batch_size": 1024,
+  "checkpoint_interval": 25000,
+  "conditioning": "history_goal_time_adaln",
+  "depth": 4,
+  "ema_decay": 0.9999,
+  "embed_dim": 192,
+  "eval_batch_size": 1024,
+  "eval_interval": 10000,
+  "exp_name": "latent_pathflow_ogbench8_cs_noisy_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "ff_dim": 2048,
+  "final_learning_rate": 1e-05,
+  "flow_sampling_steps": 16,
+  "flow_solver": "euler",
+  "goal_sampling": "hiql_uniform_future_same_trajectory",
+  "gradient_clip": 1.0,
+  "hidden_dim": 512,
+  "history_size": 3,
+  "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_cs_noisy__node3_lewm_s3072_e10_z192.h5",
+  "learning_rate": 0.0001,
+  "lewm_checkpoint_sha256": "69397269ac0783d9ba1d723e10898be8f1d907b7a3a67d4851c37f7eb4e37d14",
+  "log_interval": 1000,
+  "loss": "conditional_path_flow_matching_mse",
+  "max_goal_steps": null,
+  "num_episodes": 1000,
+  "num_heads": 8,
+  "num_rows": 1001000,
+  "num_samples": 1,
+  "num_train_episodes": 950,
+  "num_train_transitions": 950000,
+  "num_val_episodes": 50,
+  "num_val_transitions": 50000,
+  "resume": true,
+  "save_dir": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/latent_pathflow_ogbench8_cs_noisy_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "seed": 0,
+  "split_seed": 0,
+  "subgoal_steps": 10,
+  "time_dim": 64,
+  "train_fraction": 0.95,
+  "train_steps": 200000,
+  "validation_manifest": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/manifests/ogbench8_cs_noisy_h50_hist3_k10_n10000.npz",
+  "validation_manifest_metadata": {
+    "action_block": 5,
+    "format": "lewm_latent_subgoal_validation_manifest",
+    "goal_offset": 50,
+    "history_size": 3,
+    "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_cs_noisy__node3_lewm_s3072_e10_z192.h5",
+    "lewm_checkpoint_sha256": "69397269ac0783d9ba1d723e10898be8f1d907b7a3a67d4851c37f7eb4e37d14",
+    "num_episodes": 1000,
+    "num_pairs": 10000,
+    "num_rows": 1001000,
+    "sampling_seed": 1,
+    "split_seed": 0,
+    "subgoal_steps": 10,
+    "train_fraction": 0.95,
+    "validation_episode_ids": [
+      307,
+      613,
+      240,
+      561,
+      570,
+      699,
+      681,
+      936,
+      864,
+      616,
+      317,
+      780,
+      719,
+      753,
+      800,
+      568,
+      302,
+      826,
+      782,
+      73,
+      934,
+      775,
+      889,
+      115,
+      287,
+      191,
+      496,
+      104,
+      176,
+      845,
+      844,
+      7,
+      487,
+      959,
+      351,
+      581,
+      881,
+      639,
+      377,
+      78,
+      632,
+      315,
+      950,
+      29,
+      425,
+      184,
+      504,
+      801,
+      975,
+      607
+    ]
+  },
+  "validation_manifest_sha256": "da6833411fbaae95252eef22ea6b0b56acce8739e2e819fd04451845a5fe0f6f",
+  "validation_pairs": 10000,
+  "warmup_steps": 5000,
+  "weight_decay": 0.0001
+}

visual-ogbench/checkpoints/latent-path-flow/cs_play/checkpoint_200000.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8fbaeb226145eb3a33932c4f285ebe0abcf4cf70f5f8332adc76707338d93c08
+size 299898830

visual-ogbench/checkpoints/latent-path-flow/cs_play/config.json

@@ -0,0 +1,115 @@
+{
+  "action_block": 5,
+  "architecture": "latent_path_flow_transformer_encoder",
+  "batch_size": 1024,
+  "checkpoint_interval": 25000,
+  "conditioning": "history_goal_time_adaln",
+  "depth": 4,
+  "ema_decay": 0.9999,
+  "embed_dim": 192,
+  "eval_batch_size": 1024,
+  "eval_interval": 10000,
+  "exp_name": "latent_pathflow_ogbench8_cs_play_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "ff_dim": 2048,
+  "final_learning_rate": 1e-05,
+  "flow_sampling_steps": 16,
+  "flow_solver": "euler",
+  "goal_sampling": "hiql_uniform_future_same_trajectory",
+  "gradient_clip": 1.0,
+  "hidden_dim": 512,
+  "history_size": 3,
+  "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_cs_play__node3_lewm_s3072_e10_z192.h5",
+  "learning_rate": 0.0001,
+  "lewm_checkpoint_sha256": "18b71fb337ce85a720f3a708f258d2feb3a9068b1399c590848e13165263c4bc",
+  "log_interval": 1000,
+  "loss": "conditional_path_flow_matching_mse",
+  "max_goal_steps": null,
+  "num_episodes": 1000,
+  "num_heads": 8,
+  "num_rows": 1001000,
+  "num_samples": 1,
+  "num_train_episodes": 950,
+  "num_train_transitions": 950000,
+  "num_val_episodes": 50,
+  "num_val_transitions": 50000,
+  "resume": true,
+  "save_dir": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/latent_pathflow_ogbench8_cs_play_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "seed": 0,
+  "split_seed": 0,
+  "subgoal_steps": 10,
+  "time_dim": 64,
+  "train_fraction": 0.95,
+  "train_steps": 200000,
+  "validation_manifest": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/manifests/ogbench8_cs_play_h50_hist3_k10_n10000.npz",
+  "validation_manifest_metadata": {
+    "action_block": 5,
+    "format": "lewm_latent_subgoal_validation_manifest",
+    "goal_offset": 50,
+    "history_size": 3,
+    "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_cs_play__node3_lewm_s3072_e10_z192.h5",
+    "lewm_checkpoint_sha256": "18b71fb337ce85a720f3a708f258d2feb3a9068b1399c590848e13165263c4bc",
+    "num_episodes": 1000,
+    "num_pairs": 10000,
+    "num_rows": 1001000,
+    "sampling_seed": 1,
+    "split_seed": 0,
+    "subgoal_steps": 10,
+    "train_fraction": 0.95,
+    "validation_episode_ids": [
+      307,
+      613,
+      240,
+      561,
+      570,
+      699,
+      681,
+      936,
+      864,
+      616,
+      317,
+      780,
+      719,
+      753,
+      800,
+      568,
+      302,
+      826,
+      782,
+      73,
+      934,
+      775,
+      889,
+      115,
+      287,
+      191,
+      496,
+      104,
+      176,
+      845,
+      844,
+      7,
+      487,
+      959,
+      351,
+      581,
+      881,
+      639,
+      377,
+      78,
+      632,
+      315,
+      950,
+      29,
+      425,
+      184,
+      504,
+      801,
+      975,
+      607
+    ]
+  },
+  "validation_manifest_sha256": "c6a47b73969c02d348b7358bf9e841ce2bf7a09237f40f01e19d861eda4fee06",
+  "validation_pairs": 10000,
+  "warmup_steps": 5000,
+  "weight_decay": 0.0001
+}

visual-ogbench/checkpoints/latent-path-flow/ct_noisy/checkpoint_200000.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ba19b94364c8f560ee67583943e751ea62749511a34410b55e3ad3da48445711
+size 299898830

visual-ogbench/checkpoints/latent-path-flow/ct_noisy/config.json

@@ -0,0 +1,215 @@
+{
+  "action_block": 5,
+  "architecture": "latent_path_flow_transformer_encoder",
+  "batch_size": 1024,
+  "checkpoint_interval": 25000,
+  "conditioning": "history_goal_time_adaln",
+  "depth": 4,
+  "ema_decay": 0.9999,
+  "embed_dim": 192,
+  "eval_batch_size": 1024,
+  "eval_interval": 10000,
+  "exp_name": "latent_pathflow_ogbench8_ct_noisy_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "ff_dim": 2048,
+  "final_learning_rate": 1e-05,
+  "flow_sampling_steps": 16,
+  "flow_solver": "euler",
+  "goal_sampling": "hiql_uniform_future_same_trajectory",
+  "gradient_clip": 1.0,
+  "hidden_dim": 512,
+  "history_size": 3,
+  "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_ct_noisy__node3_lewm_s3072_e10_z192.h5",
+  "learning_rate": 0.0001,
+  "lewm_checkpoint_sha256": "74d078fd7d469f117cbfc0b0841a4405d019f6f0d0fbca592afeeb2b09316dd2",
+  "log_interval": 1000,
+  "loss": "conditional_path_flow_matching_mse",
+  "max_goal_steps": null,
+  "num_episodes": 3000,
+  "num_heads": 8,
+  "num_rows": 3003000,
+  "num_samples": 1,
+  "num_train_episodes": 2850,
+  "num_train_transitions": 2850000,
+  "num_val_episodes": 150,
+  "num_val_transitions": 150000,
+  "resume": true,
+  "save_dir": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/latent_pathflow_ogbench8_ct_noisy_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "seed": 0,
+  "split_seed": 0,
+  "subgoal_steps": 10,
+  "time_dim": 64,
+  "train_fraction": 0.95,
+  "train_steps": 200000,
+  "validation_manifest": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/manifests/ogbench8_ct_noisy_h50_hist3_k10_n10000.npz",
+  "validation_manifest_metadata": {
+    "action_block": 5,
+    "format": "lewm_latent_subgoal_validation_manifest",
+    "goal_offset": 50,
+    "history_size": 3,
+    "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_ct_noisy__node3_lewm_s3072_e10_z192.h5",
+    "lewm_checkpoint_sha256": "74d078fd7d469f117cbfc0b0841a4405d019f6f0d0fbca592afeeb2b09316dd2",
+    "num_episodes": 3000,
+    "num_pairs": 10000,
+    "num_rows": 3003000,
+    "sampling_seed": 1,
+    "split_seed": 0,
+    "subgoal_steps": 10,
+    "train_fraction": 0.95,
+    "validation_episode_ids": [
+      1859,
+      2336,
+      1289,
+      1111,
+      2778,
+      483,
+      706,
+      2263,
+      1938,
+      2461,
+      2064,
+      1711,
+      2665,
+      1451,
+      2389,
+      704,
+      110,
+      113,
+      2078,
+      2049,
+      2754,
+      1153,
+      2589,
+      1196,
+      1128,
+      2305,
+      2940,
+      1282,
+      2173,
+      1648,
+      563,
+      2791,
+      1840,
+      1204,
+      51,
+      1159,
+      492,
+      1556,
+      806,
+      2276,
+      434,
+      2798,
+      2871,
+      276,
+      2441,
+      546,
+      2506,
+      2843,
+      1906,
+      736,
+      57,
+      2949,
+      1433,
+      1963,
+      1178,
+      1388,
+      2121,
+      2735,
+      211,
+      1046,
+      1811,
+      1208,
+      1158,
+      2478,
+      146,
+      1346,
+      217,
+      1617,
+      2743,
+      1937,
+      2799,
+      341,
+      2928,
+      3,
+      2723,
+      2955,
+      2538,
+      1167,
+      700,
+      2462,
+      1090,
+      1699,
+      2124,
+      1899,
+      1489,
+      2985,
+      156,
+      2514,
+      1682,
+      2438,
+      835,
+      1217,
+      757,
+      2619,
+      1465,
+      1045,
+      2748,
+      2479,
+      2944,
+      1468,
+      1598,
+      301,
+      1622,
+      1343,
+      1661,
+      1306,
+      307,
+      2661,
+      240,
+      1585,
+      1594,
+      699,
+      2729,
+      1888,
+      2664,
+      1341,
+      780,
+      2767,
+      1777,
+      2848,
+      568,
+      985,
+      2350,
+      2874,
+      1097,
+      1958,
+      775,
+      889,
+      115,
+      1311,
+      2239,
+      1520,
+      2152,
+      176,
+      1869,
+      1868,
+      2055,
+      487,
+      2399,
+      2629,
+      2929,
+      639,
+      2425,
+      2680,
+      315,
+      1974,
+      29,
+      2232,
+      1825,
+      607
+    ]
+  },
+  "validation_manifest_sha256": "55596ff818a5675edfa77fcd0edbe63ff72fa76a7acd2a784517f59332c02d09",
+  "validation_pairs": 10000,
+  "warmup_steps": 5000,
+  "weight_decay": 0.0001
+}

visual-ogbench/checkpoints/latent-path-flow/ct_play/checkpoint_200000.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1c7082ffad8a68805110591d65facffd84b52584b3732811fbee804d3bd07adb
+size 299898830

visual-ogbench/checkpoints/latent-path-flow/ct_play/config.json

@@ -0,0 +1,215 @@
+{
+  "action_block": 5,
+  "architecture": "latent_path_flow_transformer_encoder",
+  "batch_size": 1024,
+  "checkpoint_interval": 25000,
+  "conditioning": "history_goal_time_adaln",
+  "depth": 4,
+  "ema_decay": 0.9999,
+  "embed_dim": 192,
+  "eval_batch_size": 1024,
+  "eval_interval": 10000,
+  "exp_name": "latent_pathflow_ogbench8_ct_play_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "ff_dim": 2048,
+  "final_learning_rate": 1e-05,
+  "flow_sampling_steps": 16,
+  "flow_solver": "euler",
+  "goal_sampling": "hiql_uniform_future_same_trajectory",
+  "gradient_clip": 1.0,
+  "hidden_dim": 512,
+  "history_size": 3,
+  "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_ct_play__node3_lewm_s3072_e10_z192.h5",
+  "learning_rate": 0.0001,
+  "lewm_checkpoint_sha256": "a534f27f1ab95e2219e1d3401ee14dbdcef375e853377dd6225aecbdb4d46fb7",
+  "log_interval": 1000,
+  "loss": "conditional_path_flow_matching_mse",
+  "max_goal_steps": null,
+  "num_episodes": 3000,
+  "num_heads": 8,
+  "num_rows": 3003000,
+  "num_samples": 1,
+  "num_train_episodes": 2850,
+  "num_train_transitions": 2850000,
+  "num_val_episodes": 150,
+  "num_val_transitions": 150000,
+  "resume": true,
+  "save_dir": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/latent_pathflow_ogbench8_ct_play_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "seed": 0,
+  "split_seed": 0,
+  "subgoal_steps": 10,
+  "time_dim": 64,
+  "train_fraction": 0.95,
+  "train_steps": 200000,
+  "validation_manifest": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/manifests/ogbench8_ct_play_h50_hist3_k10_n10000.npz",
+  "validation_manifest_metadata": {
+    "action_block": 5,
+    "format": "lewm_latent_subgoal_validation_manifest",
+    "goal_offset": 50,
+    "history_size": 3,
+    "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_ct_play__node3_lewm_s3072_e10_z192.h5",
+    "lewm_checkpoint_sha256": "a534f27f1ab95e2219e1d3401ee14dbdcef375e853377dd6225aecbdb4d46fb7",
+    "num_episodes": 3000,
+    "num_pairs": 10000,
+    "num_rows": 3003000,
+    "sampling_seed": 1,
+    "split_seed": 0,
+    "subgoal_steps": 10,
+    "train_fraction": 0.95,
+    "validation_episode_ids": [
+      1859,
+      2336,
+      1289,
+      1111,
+      2778,
+      483,
+      706,
+      2263,
+      1938,
+      2461,
+      2064,
+      1711,
+      2665,
+      1451,
+      2389,
+      704,
+      110,
+      113,
+      2078,
+      2049,
+      2754,
+      1153,
+      2589,
+      1196,
+      1128,
+      2305,
+      2940,
+      1282,
+      2173,
+      1648,
+      563,
+      2791,
+      1840,
+      1204,
+      51,
+      1159,
+      492,
+      1556,
+      806,
+      2276,
+      434,
+      2798,
+      2871,
+      276,
+      2441,
+      546,
+      2506,
+      2843,
+      1906,
+      736,
+      57,
+      2949,
+      1433,
+      1963,
+      1178,
+      1388,
+      2121,
+      2735,
+      211,
+      1046,
+      1811,
+      1208,
+      1158,
+      2478,
+      146,
+      1346,
+      217,
+      1617,
+      2743,
+      1937,
+      2799,
+      341,
+      2928,
+      3,
+      2723,
+      2955,
+      2538,
+      1167,
+      700,
+      2462,
+      1090,
+      1699,
+      2124,
+      1899,
+      1489,
+      2985,
+      156,
+      2514,
+      1682,
+      2438,
+      835,
+      1217,
+      757,
+      2619,
+      1465,
+      1045,
+      2748,
+      2479,
+      2944,
+      1468,
+      1598,
+      301,
+      1622,
+      1343,
+      1661,
+      1306,
+      307,
+      2661,
+      240,
+      1585,
+      1594,
+      699,
+      2729,
+      1888,
+      2664,
+      1341,
+      780,
+      2767,
+      1777,
+      2848,
+      568,
+      985,
+      2350,
+      2874,
+      1097,
+      1958,
+      775,
+      889,
+      115,
+      1311,
+      2239,
+      1520,
+      2152,
+      176,
+      1869,
+      1868,
+      2055,
+      487,
+      2399,
+      2629,
+      2929,
+      639,
+      2425,
+      2680,
+      315,
+      1974,
+      29,
+      2232,
+      1825,
+      607
+    ]
+  },
+  "validation_manifest_sha256": "be5c2fc1c5a29180b0caa18df6dc187e1a4eb754bb542fb9f56d19835ac1b037",
+  "validation_pairs": 10000,
+  "warmup_steps": 5000,
+  "weight_decay": 0.0001
+}

visual-ogbench/checkpoints/latent-path-flow/scene_noisy/checkpoint_200000.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:01b0d2ebb972642befb1ebee71f3c8e3a42fb5299ea0449a2a3a6acfb1ba507c
+size 299898830

visual-ogbench/checkpoints/latent-path-flow/scene_noisy/config.json

@@ -0,0 +1,115 @@
+{
+  "action_block": 5,
+  "architecture": "latent_path_flow_transformer_encoder",
+  "batch_size": 1024,
+  "checkpoint_interval": 25000,
+  "conditioning": "history_goal_time_adaln",
+  "depth": 4,
+  "ema_decay": 0.9999,
+  "embed_dim": 192,
+  "eval_batch_size": 1024,
+  "eval_interval": 10000,
+  "exp_name": "latent_pathflow_ogbench8_scene_noisy_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "ff_dim": 2048,
+  "final_learning_rate": 1e-05,
+  "flow_sampling_steps": 16,
+  "flow_solver": "euler",
+  "goal_sampling": "hiql_uniform_future_same_trajectory",
+  "gradient_clip": 1.0,
+  "hidden_dim": 512,
+  "history_size": 3,
+  "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_scene_noisy__node3_lewm_s3072_e10_z192.h5",
+  "learning_rate": 0.0001,
+  "lewm_checkpoint_sha256": "077a94328e5f1a0a815d2834f1cd833306b6beaec207771136b8adb65ee2d6da",
+  "log_interval": 1000,
+  "loss": "conditional_path_flow_matching_mse",
+  "max_goal_steps": null,
+  "num_episodes": 1000,
+  "num_heads": 8,
+  "num_rows": 1001000,
+  "num_samples": 1,
+  "num_train_episodes": 950,
+  "num_train_transitions": 950000,
+  "num_val_episodes": 50,
+  "num_val_transitions": 50000,
+  "resume": true,
+  "save_dir": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/latent_pathflow_ogbench8_scene_noisy_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "seed": 0,
+  "split_seed": 0,
+  "subgoal_steps": 10,
+  "time_dim": 64,
+  "train_fraction": 0.95,
+  "train_steps": 200000,
+  "validation_manifest": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/manifests/ogbench8_scene_noisy_h50_hist3_k10_n10000.npz",
+  "validation_manifest_metadata": {
+    "action_block": 5,
+    "format": "lewm_latent_subgoal_validation_manifest",
+    "goal_offset": 50,
+    "history_size": 3,
+    "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_scene_noisy__node3_lewm_s3072_e10_z192.h5",
+    "lewm_checkpoint_sha256": "077a94328e5f1a0a815d2834f1cd833306b6beaec207771136b8adb65ee2d6da",
+    "num_episodes": 1000,
+    "num_pairs": 10000,
+    "num_rows": 1001000,
+    "sampling_seed": 1,
+    "split_seed": 0,
+    "subgoal_steps": 10,
+    "train_fraction": 0.95,
+    "validation_episode_ids": [
+      307,
+      613,
+      240,
+      561,
+      570,
+      699,
+      681,
+      936,
+      864,
+      616,
+      317,
+      780,
+      719,
+      753,
+      800,
+      568,
+      302,
+      826,
+      782,
+      73,
+      934,
+      775,
+      889,
+      115,
+      287,
+      191,
+      496,
+      104,
+      176,
+      845,
+      844,
+      7,
+      487,
+      959,
+      351,
+      581,
+      881,
+      639,
+      377,
+      78,
+      632,
+      315,
+      950,
+      29,
+      425,
+      184,
+      504,
+      801,
+      975,
+      607
+    ]
+  },
+  "validation_manifest_sha256": "15cf7367d01db9cacf257e66c9dd04cabbccc76a41d5745e98914890fb1f6e79",
+  "validation_pairs": 10000,
+  "warmup_steps": 5000,
+  "weight_decay": 0.0001
+}

visual-ogbench/checkpoints/latent-path-flow/scene_play/checkpoint_200000.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bf1ebdb712c5e5beb52fa3948de479dc470966089b6abd7834a794ba9a653549
+size 299898830

visual-ogbench/checkpoints/latent-path-flow/scene_play/config.json

@@ -0,0 +1,115 @@
+{
+  "action_block": 5,
+  "architecture": "latent_path_flow_transformer_encoder",
+  "batch_size": 1024,
+  "checkpoint_interval": 25000,
+  "conditioning": "history_goal_time_adaln",
+  "depth": 4,
+  "ema_decay": 0.9999,
+  "embed_dim": 192,
+  "eval_batch_size": 1024,
+  "eval_interval": 10000,
+  "exp_name": "latent_pathflow_ogbench8_scene_play_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "ff_dim": 2048,
+  "final_learning_rate": 1e-05,
+  "flow_sampling_steps": 16,
+  "flow_solver": "euler",
+  "goal_sampling": "hiql_uniform_future_same_trajectory",
+  "gradient_clip": 1.0,
+  "hidden_dim": 512,
+  "history_size": 3,
+  "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_scene_play__node3_lewm_s3072_e10_z192.h5",
+  "learning_rate": 0.0001,
+  "lewm_checkpoint_sha256": "c60ca16acf0b8b597e847d6d5b26cca5aa59f576eac1788c7a0e48f634c25fa1",
+  "log_interval": 1000,
+  "loss": "conditional_path_flow_matching_mse",
+  "max_goal_steps": null,
+  "num_episodes": 1000,
+  "num_heads": 8,
+  "num_rows": 1001000,
+  "num_samples": 1,
+  "num_train_episodes": 950,
+  "num_train_transitions": 950000,
+  "num_val_episodes": 50,
+  "num_val_transitions": 50000,
+  "resume": true,
+  "save_dir": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/latent_pathflow_ogbench8_scene_play_node3lewm3072e10_hist3_sg10_ab5_uniform_ns1_n200000_b1024_s0",
+  "seed": 0,
+  "split_seed": 0,
+  "subgoal_steps": 10,
+  "time_dim": 64,
+  "train_fraction": 0.95,
+  "train_steps": 200000,
+  "validation_manifest": "/data-training/yyf/ogbench-lewm-policy-runs/latent-path-flow-ogbench8-k10/manifests/ogbench8_scene_play_h50_hist3_k10_n10000.npz",
+  "validation_manifest_metadata": {
+    "action_block": 5,
+    "format": "lewm_latent_subgoal_validation_manifest",
+    "goal_offset": 50,
+    "history_size": 3,
+    "latent_dataset": "/data-training/yyf/datasets/lewm-latents-ogbench8-node3-e10/ogbench8_scene_play__node3_lewm_s3072_e10_z192.h5",
+    "lewm_checkpoint_sha256": "c60ca16acf0b8b597e847d6d5b26cca5aa59f576eac1788c7a0e48f634c25fa1",
+    "num_episodes": 1000,
+    "num_pairs": 10000,
+    "num_rows": 1001000,
+    "sampling_seed": 1,
+    "split_seed": 0,
+    "subgoal_steps": 10,
+    "train_fraction": 0.95,
+    "validation_episode_ids": [
+      307,
+      613,
+      240,
+      561,
+      570,
+      699,
+      681,
+      936,
+      864,
+      616,
+      317,
+      780,
+      719,
+      753,
+      800,
+      568,
+      302,
+      826,
+      782,
+      73,
+      934,
+      775,
+      889,
+      115,
+      287,
+      191,
+      496,
+      104,
+      176,
+      845,
+      844,
+      7,
+      487,
+      959,
+      351,
+      581,
+      881,
+      639,
+      377,
+      78,
+      632,
+      315,
+      950,
+      29,
+      425,
+      184,
+      504,
+      801,
+      975,
+      607
+    ]
+  },
+  "validation_manifest_sha256": "831e7b4718d1d85943081daea8bcf9617417dc4e05d383821955081f49bc596d",
+  "validation_pairs": 10000,
+  "warmup_steps": 5000,
+  "weight_decay": 0.0001
+}

visual-ogbench/checkpoints/lewm/cd_noisy/config.json

@@ -0,0 +1,37 @@
+{
+  "exp_name": "lewm_ogbench8_cd_noisy_e10_bs128_s3072",
+  "dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-double-noisy-v0.npz",
+  "validation_dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-double-noisy-v0-val.npz",
+  "dataset_format": "npz",
+  "architecture": "lewm_impala_small",
+  "encoder": "impala_small",
+  "seed": 3072,
+  "epochs": 10,
+  "train_steps": null,
+  "save_interval_steps": 100000,
+  "batch_size": 128,
+  "decode_workers": 1,
+  "train_fraction": 0.9,
+  "image_size": 64,
+  "embed_dim": 192,
+  "history_size": 3,
+  "num_preds": 1,
+  "frameskip": 5,
+  "projector_hidden_dim": 2048,
+  "action_smoothed_dim": 10,
+  "action_mlp_scale": 4,
+  "predictor_depth": 6,
+  "predictor_heads": 16,
+  "predictor_dim_head": 64,
+  "predictor_mlp_dim": 2048,
+  "predictor_dropout": 0.1,
+  "predictor_emb_dropout": 0.0,
+  "learning_rate": 5e-05,
+  "lr_schedule": "optimizer_step_warmup_cosine_1pct",
+  "weight_decay": 0.001,
+  "gradient_clip": 1.0,
+  "sigreg_weight": 0.09,
+  "sigreg_knots": 17,
+  "sigreg_num_proj": 1024,
+  "precision": "bf16"
+}

visual-ogbench/checkpoints/lewm/cd_noisy/weights_epoch_10.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0f3f5f573f0246b21d0b9866e80df7027a56d6462582f78e426c921042ebc11b
+size 57212382

visual-ogbench/checkpoints/lewm/cd_play/config.json

@@ -0,0 +1,37 @@
+{
+  "exp_name": "lewm_ogbench8_cd_play_e10_bs128_s3072",
+  "dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-double-play-v0.npz",
+  "validation_dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-double-play-v0-val.npz",
+  "dataset_format": "npz",
+  "architecture": "lewm_impala_small",
+  "encoder": "impala_small",
+  "seed": 3072,
+  "epochs": 10,
+  "train_steps": null,
+  "save_interval_steps": 100000,
+  "batch_size": 128,
+  "decode_workers": 1,
+  "train_fraction": 0.9,
+  "image_size": 64,
+  "embed_dim": 192,
+  "history_size": 3,
+  "num_preds": 1,
+  "frameskip": 5,
+  "projector_hidden_dim": 2048,
+  "action_smoothed_dim": 10,
+  "action_mlp_scale": 4,
+  "predictor_depth": 6,
+  "predictor_heads": 16,
+  "predictor_dim_head": 64,
+  "predictor_mlp_dim": 2048,
+  "predictor_dropout": 0.1,
+  "predictor_emb_dropout": 0.0,
+  "learning_rate": 5e-05,
+  "lr_schedule": "optimizer_step_warmup_cosine_1pct",
+  "weight_decay": 0.001,
+  "gradient_clip": 1.0,
+  "sigreg_weight": 0.09,
+  "sigreg_knots": 17,
+  "sigreg_num_proj": 1024,
+  "precision": "bf16"
+}

visual-ogbench/checkpoints/lewm/cd_play/weights_epoch_10.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ed109fcf37df140987694809c08ab8231819a4e7932497c18c00caaf778fa183
+size 57212382

visual-ogbench/checkpoints/lewm/cs_noisy/config.json

@@ -0,0 +1,37 @@
+{
+  "exp_name": "lewm_ogbench8_cs_noisy_e10_bs128_s3072",
+  "dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-single-noisy-v0.npz",
+  "validation_dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-single-noisy-v0-val.npz",
+  "dataset_format": "npz",
+  "architecture": "lewm_impala_small",
+  "encoder": "impala_small",
+  "seed": 3072,
+  "epochs": 10,
+  "train_steps": null,
+  "save_interval_steps": 100000,
+  "batch_size": 128,
+  "decode_workers": 1,
+  "train_fraction": 0.9,
+  "image_size": 64,
+  "embed_dim": 192,
+  "history_size": 3,
+  "num_preds": 1,
+  "frameskip": 5,
+  "projector_hidden_dim": 2048,
+  "action_smoothed_dim": 10,
+  "action_mlp_scale": 4,
+  "predictor_depth": 6,
+  "predictor_heads": 16,
+  "predictor_dim_head": 64,
+  "predictor_mlp_dim": 2048,
+  "predictor_dropout": 0.1,
+  "predictor_emb_dropout": 0.0,
+  "learning_rate": 5e-05,
+  "lr_schedule": "optimizer_step_warmup_cosine_1pct",
+  "weight_decay": 0.001,
+  "gradient_clip": 1.0,
+  "sigreg_weight": 0.09,
+  "sigreg_knots": 17,
+  "sigreg_num_proj": 1024,
+  "precision": "bf16"
+}

visual-ogbench/checkpoints/lewm/cs_noisy/weights_epoch_10.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:69397269ac0783d9ba1d723e10898be8f1d907b7a3a67d4851c37f7eb4e37d14
+size 57212382

visual-ogbench/checkpoints/lewm/cs_play/config.json

@@ -0,0 +1,37 @@
+{
+  "exp_name": "lewm_ogbench8_cs_play_e10_bs128_s3072",
+  "dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-single-play-v0.npz",
+  "validation_dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-single-play-v0-val.npz",
+  "dataset_format": "npz",
+  "architecture": "lewm_impala_small",
+  "encoder": "impala_small",
+  "seed": 3072,
+  "epochs": 10,
+  "train_steps": null,
+  "save_interval_steps": 100000,
+  "batch_size": 128,
+  "decode_workers": 1,
+  "train_fraction": 0.9,
+  "image_size": 64,
+  "embed_dim": 192,
+  "history_size": 3,
+  "num_preds": 1,
+  "frameskip": 5,
+  "projector_hidden_dim": 2048,
+  "action_smoothed_dim": 10,
+  "action_mlp_scale": 4,
+  "predictor_depth": 6,
+  "predictor_heads": 16,
+  "predictor_dim_head": 64,
+  "predictor_mlp_dim": 2048,
+  "predictor_dropout": 0.1,
+  "predictor_emb_dropout": 0.0,
+  "learning_rate": 5e-05,
+  "lr_schedule": "optimizer_step_warmup_cosine_1pct",
+  "weight_decay": 0.001,
+  "gradient_clip": 1.0,
+  "sigreg_weight": 0.09,
+  "sigreg_knots": 17,
+  "sigreg_num_proj": 1024,
+  "precision": "bf16"
+}

visual-ogbench/checkpoints/lewm/cs_play/weights_epoch_10.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18b71fb337ce85a720f3a708f258d2feb3a9068b1399c590848e13165263c4bc
+size 57212382

visual-ogbench/checkpoints/lewm/ct_noisy/config.json

@@ -0,0 +1,37 @@
+{
+  "exp_name": "lewm_ogbench8_ct_noisy_e10_bs128_s3072",
+  "dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-triple-noisy-v0.npz",
+  "validation_dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-triple-noisy-v0-val.npz",
+  "dataset_format": "npz",
+  "architecture": "lewm_impala_small",
+  "encoder": "impala_small",
+  "seed": 3072,
+  "epochs": 10,
+  "train_steps": null,
+  "save_interval_steps": 100000,
+  "batch_size": 128,
+  "decode_workers": 1,
+  "train_fraction": 0.9,
+  "image_size": 64,
+  "embed_dim": 192,
+  "history_size": 3,
+  "num_preds": 1,
+  "frameskip": 5,
+  "projector_hidden_dim": 2048,
+  "action_smoothed_dim": 10,
+  "action_mlp_scale": 4,
+  "predictor_depth": 6,
+  "predictor_heads": 16,
+  "predictor_dim_head": 64,
+  "predictor_mlp_dim": 2048,
+  "predictor_dropout": 0.1,
+  "predictor_emb_dropout": 0.0,
+  "learning_rate": 5e-05,
+  "lr_schedule": "optimizer_step_warmup_cosine_1pct",
+  "weight_decay": 0.001,
+  "gradient_clip": 1.0,
+  "sigreg_weight": 0.09,
+  "sigreg_knots": 17,
+  "sigreg_num_proj": 1024,
+  "precision": "bf16"
+}

visual-ogbench/checkpoints/lewm/ct_noisy/weights_epoch_10.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:74d078fd7d469f117cbfc0b0841a4405d019f6f0d0fbca592afeeb2b09316dd2
+size 57212382

visual-ogbench/checkpoints/lewm/ct_play/config.json

@@ -0,0 +1,37 @@
+{
+  "exp_name": "lewm_ogbench8_ct_play_e10_bs128_s3072",
+  "dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-triple-play-v0.npz",
+  "validation_dataset_path": "/data-training/yyf/ogbench-cache/data/visual-cube-triple-play-v0-val.npz",
+  "dataset_format": "npz",
+  "architecture": "lewm_impala_small",
+  "encoder": "impala_small",
+  "seed": 3072,
+  "epochs": 10,
+  "train_steps": null,
+  "save_interval_steps": 100000,
+  "batch_size": 128,
+  "decode_workers": 1,
+  "train_fraction": 0.9,
+  "image_size": 64,
+  "embed_dim": 192,
+  "history_size": 3,
+  "num_preds": 1,
+  "frameskip": 5,
+  "projector_hidden_dim": 2048,
+  "action_smoothed_dim": 10,
+  "action_mlp_scale": 4,
+  "predictor_depth": 6,
+  "predictor_heads": 16,
+  "predictor_dim_head": 64,
+  "predictor_mlp_dim": 2048,
+  "predictor_dropout": 0.1,
+  "predictor_emb_dropout": 0.0,
+  "learning_rate": 5e-05,
+  "lr_schedule": "optimizer_step_warmup_cosine_1pct",
+  "weight_decay": 0.001,
+  "gradient_clip": 1.0,
+  "sigreg_weight": 0.09,
+  "sigreg_knots": 17,
+  "sigreg_num_proj": 1024,
+  "precision": "bf16"
+}

visual-ogbench/checkpoints/lewm/ct_play/weights_epoch_10.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a534f27f1ab95e2219e1d3401ee14dbdcef375e853377dd6225aecbdb4d46fb7
+size 57212382

visual-ogbench/checkpoints/lewm/scene_noisy/config.json

@@ -0,0 +1,37 @@
+{
+  "exp_name": "lewm_ogbench8_scene_noisy_e10_bs128_s3072",
+  "dataset_path": "/data-training/yyf/ogbench-cache/data/visual-scene-noisy-v0.npz",
+  "validation_dataset_path": "/data-training/yyf/ogbench-cache/data/visual-scene-noisy-v0-val.npz",
+  "dataset_format": "npz",
+  "architecture": "lewm_impala_small",
+  "encoder": "impala_small",
+  "seed": 3072,
+  "epochs": 10,
+  "train_steps": null,
+  "save_interval_steps": 100000,
+  "batch_size": 128,
+  "decode_workers": 1,
+  "train_fraction": 0.9,
+  "image_size": 64,
+  "embed_dim": 192,
+  "history_size": 3,
+  "num_preds": 1,
+  "frameskip": 5,
+  "projector_hidden_dim": 2048,
+  "action_smoothed_dim": 10,
+  "action_mlp_scale": 4,
+  "predictor_depth": 6,
+  "predictor_heads": 16,
+  "predictor_dim_head": 64,
+  "predictor_mlp_dim": 2048,
+  "predictor_dropout": 0.1,
+  "predictor_emb_dropout": 0.0,
+  "learning_rate": 5e-05,
+  "lr_schedule": "optimizer_step_warmup_cosine_1pct",
+  "weight_decay": 0.001,
+  "gradient_clip": 1.0,
+  "sigreg_weight": 0.09,
+  "sigreg_knots": 17,
+  "sigreg_num_proj": 1024,
+  "precision": "bf16"
+}

visual-ogbench/checkpoints/lewm/scene_noisy/weights_epoch_10.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:077a94328e5f1a0a815d2834f1cd833306b6beaec207771136b8adb65ee2d6da
+size 57212382

visual-ogbench/checkpoints/lewm/scene_play/config.json

@@ -0,0 +1,37 @@
+{
+  "exp_name": "lewm_ogbench8_scene_play_e10_bs128_s3072",
+  "dataset_path": "/data-training/yyf/ogbench-cache/data/visual-scene-play-v0.npz",
+  "validation_dataset_path": "/data-training/yyf/ogbench-cache/data/visual-scene-play-v0-val.npz",
+  "dataset_format": "npz",
+  "architecture": "lewm_impala_small",
+  "encoder": "impala_small",
+  "seed": 3072,
+  "epochs": 10,
+  "train_steps": null,
+  "save_interval_steps": 100000,
+  "batch_size": 128,
+  "decode_workers": 1,
+  "train_fraction": 0.9,
+  "image_size": 64,
+  "embed_dim": 192,
+  "history_size": 3,
+  "num_preds": 1,
+  "frameskip": 5,
+  "projector_hidden_dim": 2048,
+  "action_smoothed_dim": 10,
+  "action_mlp_scale": 4,
+  "predictor_depth": 6,
+  "predictor_heads": 16,
+  "predictor_dim_head": 64,
+  "predictor_mlp_dim": 2048,
+  "predictor_dropout": 0.1,
+  "predictor_emb_dropout": 0.0,
+  "learning_rate": 5e-05,
+  "lr_schedule": "optimizer_step_warmup_cosine_1pct",
+  "weight_decay": 0.001,
+  "gradient_clip": 1.0,
+  "sigreg_weight": 0.09,
+  "sigreg_knots": 17,
+  "sigreg_num_proj": 1024,
+  "precision": "bf16"
+}

visual-ogbench/checkpoints/lewm/scene_play/weights_epoch_10.msgpack

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c60ca16acf0b8b597e847d6d5b26cca5aa59f576eac1788c7a0e48f634c25fa1
+size 57212382