DavidSunok's picture
docs: organize current INTACT checkpoint releases
d22b694 verified
Raw History Blame Contribute Delete
9.55 kB
{
"schema_version": 1,
"release": {
"provider": "huggingface",
"repository": "INTACT-JEPA/INTACT",
"revision": "main",
"base_code_commit": "235b6a3a92db4d0f1b3a40597ab1f407db4fd15b",
"path": "INTACT-no-previous-action",
"source_revision": "intact-goal-e1-single-history-free-v1"
},
"model": {
"name": "History-free goal-displacement INTACT single-task reproduction",
"epoch": 1,
"shared_encoder": false,
"feature_layout": "four_slot",
"actor_latent_tensor_slots": 3,
"actor_input": "[z_t, m_t, z_t * m_t]",
"uses_previous_action": false,
"window": "7/7",
"terminal_goal": "z7"
},
"scientific_contract": {
"method": "INTACT",
"intent": "goal_displacement",
"num_frames": 8,
"local_indices": "0..6",
"goal_indices": "0..6",
"terminal_goal": "z7",
"epochs": 1,
"batch_size": 256,
"num_workers": 6,
"precision": "bf16-mixed",
"sdpa": "MATH",
"optimizer": "AdamW",
"learning_rate": 0.0005,
"weight_decay": 0.001,
"sigreg_weight": 0.02,
"sigreg_num_proj": 1024,
"training_seeds": [
0,
42,
3072
],
"evaluation_seeds": [
0,
1,
42
],
"evaluation_protocol": "Official Direct, H=5, RH=5, zero search"
},
"training_seeds": [
{
"seed": 0,
"shards": [
{
"task": "pusht",
"path": "checkpoints/intact_goal_history_free_pusht_s0_p77_v1/weights_epoch_1.pt",
"bytes": 83161344,
"sha256": "a90b17296f1f4fb53771b5d0abb1aeb5f4d9b63d39aed9a893f1c71f60b5a984",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 87.0,
"1": 82.0,
"42": 86.0
},
"official_direct_mean_sr_percent": 85.0
},
{
"task": "cube",
"path": "checkpoints/intact_goal_history_free_cube_s0_p77_v1/weights_epoch_1.pt",
"bytes": 83332608,
"sha256": "d387aa5771f8fa077ed9127ca87801911ae342bd39bd2de256235ed18e738ab0",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 99.0,
"1": 100.0,
"42": 100.0
},
"official_direct_mean_sr_percent": 99.66666666666667
},
{
"task": "reacher",
"path": "checkpoints/intact_goal_history_free_reacher_s0_p77_v1/weights_epoch_1.pt",
"bytes": 83161344,
"sha256": "a51968d11733b8478cf524dedcbbcb430db9f6ea178fb2111d925db8b51c0ec4",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 93.0,
"1": 93.0,
"42": 93.0
},
"official_direct_mean_sr_percent": 93.0
},
{
"task": "tworoom",
"path": "checkpoints/intact_goal_history_free_tworoom_s0_p77_v1/weights_epoch_1.pt",
"bytes": 83161344,
"sha256": "4e773fbad91ff8c649ec7adeee9edf1accc83049a9c4fed38a66ad3a52ee5be0",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 100.0,
"1": 97.0,
"42": 100.0
},
"official_direct_mean_sr_percent": 99.0
}
],
"official_direct_sr_percent": {
"pusht": 85.0,
"cube": 99.66666666666667,
"reacher": 93.0,
"tworoom": 99.0,
"macro": 94.16666666666667
},
"asset": "intact-goal-e1-single-history-free-seed0.tar.gz",
"asset_bytes": 307569269,
"asset_sha256": "73c28e91592c87f286a010c069897d755cb84eef0e521da10a4f72440df1f58f"
},
{
"seed": 42,
"shards": [
{
"task": "pusht",
"path": "checkpoints/intact_goal_history_free_pusht_s42_p77_v1/weights_epoch_1.pt",
"bytes": 83161344,
"sha256": "79007aec7435080149b96ff15e05955db180a4bac14c1cc5b7d256a9f8fcfc22",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 86.0,
"1": 86.0,
"42": 86.0
},
"official_direct_mean_sr_percent": 86.0
},
{
"task": "cube",
"path": "checkpoints/intact_goal_history_free_cube_s42_p77_v1/weights_epoch_1.pt",
"bytes": 83332608,
"sha256": "f2ff432b049ae78be44a93510d9e4a6a0d429fe55bfc7d02a0f74a595f7bfe82",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 99.0,
"1": 100.0,
"42": 100.0
},
"official_direct_mean_sr_percent": 99.66666666666667
},
{
"task": "reacher",
"path": "checkpoints/intact_goal_history_free_reacher_s42_p77_v1/weights_epoch_1.pt",
"bytes": 83161344,
"sha256": "52ca8bff279cc153ea24c7c9219dd57523d2c079006b1779dbbc60cf1f4f3834",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 92.0,
"1": 94.0,
"42": 95.0
},
"official_direct_mean_sr_percent": 93.66666666666667
},
{
"task": "tworoom",
"path": "checkpoints/intact_goal_history_free_tworoom_s42_p77_v1/weights_epoch_1.pt",
"bytes": 83161344,
"sha256": "41d1a76bcd2e4f08ff1bb9bcae72f7de4de763395fa81f7fa8d340865ad23f1d",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 97.0,
"1": 98.0,
"42": 99.0
},
"official_direct_mean_sr_percent": 98.0
}
],
"official_direct_sr_percent": {
"pusht": 86.0,
"cube": 99.66666666666667,
"reacher": 93.66666666666667,
"tworoom": 98.0,
"macro": 94.33333333333334
},
"asset": "intact-goal-e1-single-history-free-seed42.tar.gz",
"asset_bytes": 307570839,
"asset_sha256": "125094f39918fb5ee2548e0624bef4346334020701112ee19561fd31dae2cff9"
},
{
"seed": 3072,
"shards": [
{
"task": "pusht",
"path": "checkpoints/intact_goal_history_free_pusht_s3072_p77_v1/weights_epoch_1.pt",
"bytes": 83161344,
"sha256": "bb4377be9a5d2923f36d7ca1936c38fd3a530c103e588e9af01f26b2e230de34",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 86.0,
"1": 85.0,
"42": 84.0
},
"official_direct_mean_sr_percent": 85.0
},
{
"task": "cube",
"path": "checkpoints/intact_goal_history_free_cube_s3072_p77_v1/weights_epoch_1.pt",
"bytes": 83332608,
"sha256": "da661dc8700535c95825e1f9a74c4371e04da2342652204c99ee992f6f8310b3",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 97.0,
"1": 100.0,
"42": 100.0
},
"official_direct_mean_sr_percent": 99.0
},
{
"task": "reacher",
"path": "checkpoints/intact_goal_history_free_reacher_s3072_p77_v1/weights_epoch_1.pt",
"bytes": 83161344,
"sha256": "47fa828348b9b26587aa5b82e9f3d15bcee41c14d0f9f34f38450c5445ec738b",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 93.0,
"1": 94.0,
"42": 94.0
},
"official_direct_mean_sr_percent": 93.66666666666667
},
{
"task": "tworoom",
"path": "checkpoints/intact_goal_history_free_tworoom_s3072_p77_v1/weights_epoch_1.pt",
"bytes": 83161344,
"sha256": "5767c633b0dcea281794da53facc3b9cb11b6625f1587ec56ebe54b43259069a",
"tensors": 317,
"actor_input_features": 576,
"official_direct_sr_percent": {
"0": 100.0,
"1": 98.0,
"42": 100.0
},
"official_direct_mean_sr_percent": 99.33333333333333
}
],
"official_direct_sr_percent": {
"pusht": 85.0,
"cube": 99.0,
"reacher": 93.66666666666667,
"tworoom": 99.33333333333333,
"macro": 94.25
},
"asset": "intact-goal-e1-single-history-free-seed3072.tar.gz",
"asset_bytes": 307573590,
"asset_sha256": "5d9c496e6701b302b20622eb7c668e4242853e4f33f0a9ef9bdab50f2fee5180"
}
],
"evaluation": {
"interface": "Official Direct",
"horizon": 5,
"replan_horizon": 5,
"candidate_sequences": 0,
"evaluation_seeds": [
0,
1,
42
],
"episodes_per_evaluation_seed": 100,
"evaluation_count": 36,
"episodes": 3600,
"status": "PASS",
"overall_successes": 3393,
"overall_success_rate": 94.25,
"task_mean_sr_percent": {
"pusht": 85.33333333333333,
"cube": 99.44444444444444,
"reacher": 93.44444444444444,
"tworoom": 98.77777777777777
},
"task_train_seed_sample_std_percent": {
"pusht": 0.5773502691896257,
"cube": 0.38490017945975324,
"reacher": 0.38490017945975324,
"tworoom": 0.6938886664887091
},
"macro_mean_sr_percent": 94.25,
"macro_train_seed_sample_std_percent": 0.0833333333333357
}
}