Shuaijun's picture
Upload 38 files
f3d43a3 verified
Raw History Blame Contribute Delete
4.42 kB
{
"project": "D-JEPA",
"profile": "pusht-predictive-adaptation",
"source_sha256": "ccbd6ca30c550b60a186c5d0a1fa5f1a43b7a08b704b19037586ff3de52b0887",
"weights_sha256": "b2bcb7880fb9477a7bb9aaa3c3380b66e4cd5e1767a4093f990c72766ff8e49a",
"tensor_count": 22,
"public_status": "local_release_candidate_license_not_yet_selected",
"architecture": "predictor_patch",
"upstream_included": false,
"base_checkpoint_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
"lineage": {
"method": "dtail_jepa_p_v2",
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
"tdjepa_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
"refiner_sha256": "84c2b0fb93afdad9279b220ecf696642528899bb233a21ab1489c31ec7359d9f",
"refiner_lineage": {
"method": "dtail_jepa_r_v2",
"pool_sha256": "f3adae4e85c033465a7b663064503f1808dde7fa95a05ebe20da8c7f8db18dba",
"lewm_weights_sha256": "5487ac45812ded38e11472eb7084d28c81c18e85154293125dd51c278fcd72c9",
"tdjepa_weights_sha256": "328db2682a311a70bc4d81b4394fcf9ac1eb8369c827dfe199283efcbaeeac80",
"fusion_alpha": 0.42,
"fit_start_count": 448,
"calibration_start_count": 64,
"best_update": 550
},
"trainable_names": [
"predictor.transformer.layers.5.attn.norm.weight",
"predictor.transformer.layers.5.attn.norm.bias",
"predictor.transformer.layers.5.attn.to_qkv.weight",
"predictor.transformer.layers.5.attn.to_out.0.weight",
"predictor.transformer.layers.5.attn.to_out.0.bias",
"predictor.transformer.layers.5.mlp.net.0.weight",
"predictor.transformer.layers.5.mlp.net.0.bias",
"predictor.transformer.layers.5.mlp.net.1.weight",
"predictor.transformer.layers.5.mlp.net.1.bias",
"predictor.transformer.layers.5.mlp.net.4.weight",
"predictor.transformer.layers.5.mlp.net.4.bias",
"predictor.transformer.layers.5.adaLN_modulation.1.weight",
"predictor.transformer.layers.5.adaLN_modulation.1.bias",
"pred_proj.net.0.weight",
"pred_proj.net.0.bias",
"pred_proj.net.1.weight",
"pred_proj.net.1.bias",
"pred_proj.net.3.weight",
"pred_proj.net.3.bias"
],
"best_update": 200,
"fit_start_count": 384,
"calibration_start_count": 128,
"fusion_alpha": 0.42,
"config_digest": "c65cc25c5968eb762c77ec6069170551d20119f2cfa71ddd4beab795be167007",
"code_sha256": {
"dcv/dtail_v2_plastic.py": "09f0140772ea98962fcc78a2383d2d8d309b3e55bcea71f3c1ff28af15b460a9",
"dcv/dtail_v2_plastic_data.py": "514feac427ce8318f5199eed209449ec87cf730b60bbf02e20e09edac9c1b115",
"dcv/train_dtail_v2_plastic.py": "1e3da1d9b2608dd1509efdcb5a4e248a8ea4498f921c9052909474088183c1c5",
"dcv/run_dtail_v2_plastic_experiment.py": "a90d5547d9cda4db9ce4fd604b4e6bbcd6bfdaae0c9119b364adfc7b42165aee",
"dcv/dtail_v2_refine.py": "c1339ea569efd1547a024c06f0782a51b5fd3094dd7ca19dd1c4c2fd13b538cd",
"dcv/dtail_v2_fusion.py": "cb8b908d75d5bf09fdfbe935759571e387cd2dc76ec4a4d9b6bb9d0c261671c7",
"dcv/dtail_v2_data.py": "73f1c3b1a0dcb6f9c6357ab0b0a5dc2820b8a8651234c0be0dcf2593cdf4891d",
"dcv/run_dtail_v2.py": "a32c14b47801e6f5b7a1b11eca3b8fb57eccb5fc33f8a5f49ccd5ca2c83a15f8",
"dcv/hard_suite_reader.py": "8f12d1eaa6aedf439a038e98dad48e2f9776d1abc4b8dced89dd10033cf50f4e",
"dcv/hard_suite_runtime_inputs.py": "80cef66da9b554a55191f6474ef062b9b8e0202eefcd73ba0038deed8c598578",
"closure_jepa/rollout.py": "b671a527c18f25c432b3cf47fc2bdd0aba44e0b5b878ec19fd3f2814bb908538",
"closure_jepa/tdjepa_pool_adapter.py": "157ba3b9500e4818607b1225648bd2f08ee367dbc995b7f5a69bceb07433d8a8",
"closure_jepa/tdjepa_runtime.py": "eff63651f17a867ae091da53bdb93c597fcab20c099cfbaaf5d22b46431465aa"
},
"runtime_action_inputs": {
"artifact_sha256": "4187ade9296ff98534b335304035ede51c9c2d488f11e3993bff7ae0f3adc143",
"sidecar_sha256": "a1d78eac4ac03320a16d1c7f790dafb4c8291544b0cee0bab7044e23acfff8fe",
"validation_report_sha256": "c880526c322936f6da3b35f2c97c3b0d7b287d24553a6ff9f01d5b1a6061ae43",
"completion_marker_path": "source-file:pusht_dtail_runtime_actions_v1_1.npz.complete.json"
}
},
"usage": "Load exact pretrained Temporal-Distance-JEPA then replace these parameters. This is a parameter patch, not a full predictor."
}