Add files using upload-large-folder tool
Browse files- config.yaml +23 -16
- snapshot_100.pt +3 -0
- snapshot_120.pt +3 -0
- snapshot_125.pt +3 -0
- snapshot_140.pt +3 -0
- snapshot_160.pt +3 -0
- snapshot_175.pt +3 -0
- snapshot_180.pt +3 -0
- snapshot_20.pt +3 -0
- snapshot_200.pt +3 -0
- snapshot_220.pt +3 -0
- snapshot_240.pt +3 -0
- snapshot_250.pt +3 -0
- snapshot_260.pt +3 -0
- snapshot_300.pt +3 -0
- snapshot_350.pt +3 -0
- snapshot_40.pt +3 -0
- snapshot_400.pt +3 -0
- snapshot_450.pt +3 -0
- snapshot_50.pt +3 -0
- snapshot_500.pt +3 -0
- snapshot_60.pt +3 -0
- snapshot_75.pt +3 -0
- snapshot_80.pt +3 -0
- snapshot_last.pt +3 -0
config.yaml
CHANGED
|
@@ -13,7 +13,7 @@ save_train_video: true
|
|
| 13 |
use_tb: true
|
| 14 |
use_wandb: true
|
| 15 |
wandb_run_id: '36313_0'
|
| 16 |
-
wandb_notes:
|
| 17 |
eval: false
|
| 18 |
eval_on_sim: true
|
| 19 |
simulation: true
|
|
@@ -115,7 +115,7 @@ snapshot_root_dir: /mnt/grasp_high_usage/leonmkim/contact_estimation/FISH
|
|
| 115 |
save_snapshot: true
|
| 116 |
save_last_snapshot: true
|
| 117 |
save_snapshot_when_done: true
|
| 118 |
-
top_k_checkpoints:
|
| 119 |
save_snapshot_link_to_weights_dir: deprecated
|
| 120 |
restore_missing_checkpoint: null
|
| 121 |
bc_regularize: false
|
|
@@ -309,6 +309,8 @@ agent:
|
|
| 309 |
contact_model_name: ${contact_model_name}
|
| 310 |
zero_centered: false
|
| 311 |
crop_distractors_tlhw: ${agent.config.policy_cfg.crop_distractors_tlhw}
|
|
|
|
|
|
|
| 312 |
suite:
|
| 313 |
suite: frankagym
|
| 314 |
name: frankagym
|
|
@@ -319,15 +321,15 @@ suite:
|
|
| 319 |
num_train_frames: 1000000
|
| 320 |
num_seed_frames: 0
|
| 321 |
num_train_epochs: 500
|
| 322 |
-
validate_every_epochs:
|
| 323 |
-
eval_on_sim_every_epochs:
|
| 324 |
-
validate_diffusion_on_action_loss_every_epochs:
|
| 325 |
eval_on_sim_every_epochs_diffusion: 500
|
| 326 |
-
train_eval_diffusion_on_action_loss_every_epochs:
|
| 327 |
-
check_topk_every_epochs:
|
| 328 |
-
save_snapshot_every_epochs:
|
| 329 |
-
num_seen_eval_envs:
|
| 330 |
-
num_unseen_eval_envs:
|
| 331 |
evals_per_env: 1
|
| 332 |
eval_every_frames: 2000
|
| 333 |
num_eval_episodes: 5
|
|
@@ -525,6 +527,8 @@ suite:
|
|
| 525 |
contact_model_name: ${contact_model_name}
|
| 526 |
zero_centered: false
|
| 527 |
crop_distractors_tlhw: ${agent.config.policy_cfg.crop_distractors_tlhw}
|
|
|
|
|
|
|
| 528 |
true_action_history: ${true_action_history}
|
| 529 |
num_train_frames_bc: 50000
|
| 530 |
num_train_frames_drq: 1100000
|
|
@@ -535,13 +539,13 @@ num_train_frames_diffusion: 1000000
|
|
| 535 |
num_train_epochs_bc: 5000
|
| 536 |
num_train_epochs_diffusion: 500
|
| 537 |
validate_every_epochs_bc: 5
|
| 538 |
-
validate_every_epochs_diffusion:
|
| 539 |
-
validate_diffusion_on_action_loss_every_epochs:
|
| 540 |
-
train_eval_diffusion_on_action_loss_every_epochs:
|
| 541 |
-
eval_on_sim_every_epochs_diffusion:
|
| 542 |
check_topk_every_epochs: 5
|
| 543 |
-
check_topk_every_epochs_diffusion:
|
| 544 |
-
save_snapshot_every_epochs_diffusion:
|
| 545 |
x_limit:
|
| 546 |
- 0.2
|
| 547 |
- 0.7
|
|
@@ -585,3 +589,6 @@ test:
|
|
| 585 |
local_snapshot_dir: ${test.local_snapshot_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}
|
| 586 |
resume_wandb_run: false
|
| 587 |
feature_type: 180x240_1_RGB_D_2.0_cntct_1.0_gt_contact_acthist_hst4_out32_dwnkrnl3_dwnstrd2_dwnpd1
|
|
|
|
|
|
|
|
|
|
|
|
| 13 |
use_tb: true
|
| 14 |
use_wandb: true
|
| 15 |
wandb_run_id: '36313_0'
|
| 16 |
+
wandb_notes: 36313_0_req_50127_0
|
| 17 |
eval: false
|
| 18 |
eval_on_sim: true
|
| 19 |
simulation: true
|
|
|
|
| 115 |
save_snapshot: true
|
| 116 |
save_last_snapshot: true
|
| 117 |
save_snapshot_when_done: true
|
| 118 |
+
top_k_checkpoints: 10
|
| 119 |
save_snapshot_link_to_weights_dir: deprecated
|
| 120 |
restore_missing_checkpoint: null
|
| 121 |
bc_regularize: false
|
|
|
|
| 309 |
contact_model_name: ${contact_model_name}
|
| 310 |
zero_centered: false
|
| 311 |
crop_distractors_tlhw: ${agent.config.policy_cfg.crop_distractors_tlhw}
|
| 312 |
+
deterministic_actions: true
|
| 313 |
+
deterministic_actions_seed: 0
|
| 314 |
suite:
|
| 315 |
suite: frankagym
|
| 316 |
name: frankagym
|
|
|
|
| 321 |
num_train_frames: 1000000
|
| 322 |
num_seed_frames: 0
|
| 323 |
num_train_epochs: 500
|
| 324 |
+
validate_every_epochs: 20
|
| 325 |
+
eval_on_sim_every_epochs: 20
|
| 326 |
+
validate_diffusion_on_action_loss_every_epochs: 20
|
| 327 |
eval_on_sim_every_epochs_diffusion: 500
|
| 328 |
+
train_eval_diffusion_on_action_loss_every_epochs: 20
|
| 329 |
+
check_topk_every_epochs: 20
|
| 330 |
+
save_snapshot_every_epochs: 20
|
| 331 |
+
num_seen_eval_envs: 0
|
| 332 |
+
num_unseen_eval_envs: 100
|
| 333 |
evals_per_env: 1
|
| 334 |
eval_every_frames: 2000
|
| 335 |
num_eval_episodes: 5
|
|
|
|
| 527 |
contact_model_name: ${contact_model_name}
|
| 528 |
zero_centered: false
|
| 529 |
crop_distractors_tlhw: ${agent.config.policy_cfg.crop_distractors_tlhw}
|
| 530 |
+
deterministic_actions: true
|
| 531 |
+
deterministic_actions_seed: 0
|
| 532 |
true_action_history: ${true_action_history}
|
| 533 |
num_train_frames_bc: 50000
|
| 534 |
num_train_frames_drq: 1100000
|
|
|
|
| 539 |
num_train_epochs_bc: 5000
|
| 540 |
num_train_epochs_diffusion: 500
|
| 541 |
validate_every_epochs_bc: 5
|
| 542 |
+
validate_every_epochs_diffusion: 20
|
| 543 |
+
validate_diffusion_on_action_loss_every_epochs: 20
|
| 544 |
+
train_eval_diffusion_on_action_loss_every_epochs: 20
|
| 545 |
+
eval_on_sim_every_epochs_diffusion: 20
|
| 546 |
check_topk_every_epochs: 5
|
| 547 |
+
check_topk_every_epochs_diffusion: 20
|
| 548 |
+
save_snapshot_every_epochs_diffusion: 40
|
| 549 |
x_limit:
|
| 550 |
- 0.2
|
| 551 |
- 0.7
|
|
|
|
| 589 |
local_snapshot_dir: ${test.local_snapshot_root_dir}/exp_local/${suite.name}_${obs_type}/${task_name}
|
| 590 |
resume_wandb_run: false
|
| 591 |
feature_type: 180x240_1_RGB_D_2.0_cntct_1.0_gt_contact_acthist_hst4_out32_dwnkrnl3_dwnstrd2_dwnpd1
|
| 592 |
+
eval_on_sim_past_checkpoints: false
|
| 593 |
+
universal_unseen_env_seed_start: 1000000
|
| 594 |
+
restore_and_eval_all_missing_checkpoints: true
|
snapshot_100.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0f385d6843d495fdce187d5385466c6ac9510b9a6a2062bcef220950627251c5
|
| 3 |
+
size 192469160
|
snapshot_120.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1bab763451c162663680ee752664c16ef6f5ebb24df7f5ba12352bc94c6dcc7e
|
| 3 |
+
size 192476069
|
snapshot_125.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5bf7f4617e644c7e7315bf7cb841e2ef4d9ff879120c34b9fc14be1c9e3e9504
|
| 3 |
+
size 192469288
|
snapshot_140.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bce8a6e2aba0ae994682eac6012712502b0ec76295d54c130b724578b0dfce8e
|
| 3 |
+
size 192476261
|
snapshot_160.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1d5b069ecc781897a6e19742980f105ff414897e180cbe60e0e429321d1ae7fe
|
| 3 |
+
size 192476261
|
snapshot_175.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:11bf86b30af10017d819a5e3c19b1f73f60a8468a6da9811f7a652f5891fa46e
|
| 3 |
+
size 192469798
|
snapshot_180.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:863f47a9f459ca9dbd5fb284619a75a51bc03d580f27a75ab312208af964ead1
|
| 3 |
+
size 192476261
|
snapshot_20.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:08234548305ddb9cdb1d453fdce8775598a1aefb7de6d269d09aa1df833dfa49
|
| 3 |
+
size 192475047
|
snapshot_200.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ca609e01ce388a53f30240cf05ca23f0b8d9a87cedc051dcb851331fef31340
|
| 3 |
+
size 192469798
|
snapshot_220.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c64466db0028c0d4e58d894c3729a4925786106770eefa2a087e7ece00e79ae7
|
| 3 |
+
size 192476261
|
snapshot_240.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:94e66aa7796234a991a52053ed0113e78dfd3b864041392f0aa129ca5f331982
|
| 3 |
+
size 192476645
|
snapshot_250.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c7c6ecb234bb07c2b4dbddb75bd9c364dd9047f70091badfac026a4429cc3d16
|
| 3 |
+
size 192469798
|
snapshot_260.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ebfc3193b880a4eb52499f0d389c9f819dc6bc8b3ea5b6a34ef46999d82ac3c
|
| 3 |
+
size 192476517
|
snapshot_300.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bb0828b6f5db9e33aa32925531fd3452cd2ecf12ef372213c48ee555475ee9f0
|
| 3 |
+
size 192469798
|
snapshot_350.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e84d3fd37042189c52c166d7b2745c1655cfdf82bc7be463af8b21db3f7f6c29
|
| 3 |
+
size 192469798
|
snapshot_40.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:0b74478f35344aaacc30d1950f56e06348a17f53354a7f9e1289421206f66895
|
| 3 |
+
size 192475685
|
snapshot_400.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:35a9be4b0ef80dc069e59bab322b84107063ec6fcf0bb6c50437f353362e3934
|
| 3 |
+
size 192469798
|
snapshot_450.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c7833c654609b957821e21cd57dc2dbc295f3b2b3a7b05fe63aa991fef192186
|
| 3 |
+
size 192469798
|
snapshot_50.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8ff2274e3f533061539533c3631c4b663c18e1f398107f9b856f9508a8e4beac
|
| 3 |
+
size 192468840
|
snapshot_500.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:836c5d04e1e1a4f54d1c91c7bb27c9c3dd2cd4697713b9117415a860dd39d6a6
|
| 3 |
+
size 192469798
|
snapshot_60.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a367ececcd8f74a0bffd9a2fd1697121de50c1de79c91463ef8030db06848e91
|
| 3 |
+
size 192475813
|
snapshot_75.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:197ade40d2bab4a71e2a87736cfdcbaa2194688465d4fb480cd426cfe52f0656
|
| 3 |
+
size 192469032
|
snapshot_80.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a94dab827ab1179254c04b8f93b923848c0c319b423f262e89ff07769870386d
|
| 3 |
+
size 192475941
|
snapshot_last.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4ebfc3193b880a4eb52499f0d389c9f819dc6bc8b3ea5b6a34ef46999d82ac3c
|
| 3 |
+
size 192476517
|