Echo-Memory / checkpoints.json
Wayne-King's picture
Release Echo-Memory Wan 2.1 1.3B memory baseline checkpoints (epoch-0)
886e247 verified
{
"repo_id": "Echo-Team/Echo-Memory",
"backbone": "Wan2.1-T2V-1.3B",
"checkpoints": [
{
"id": "context_k1",
"family": "Raw context",
"paper_row": "Context K=1",
"train_recipe": "train/context_learning/run_pre_qkv_ctx1.sh",
"hf_path": "context_k1/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_from_scratch_merged_cam_ctx_1_noise_15_atomic_cam_inject_pre_qkv_single_rt",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
},
{
"id": "context_k20",
"family": "Raw context",
"paper_row": "Context K=20",
"train_recipe": "train/context_learning/run_pre_qkv_ctx20.sh",
"hf_path": "context_k20/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_from_scratch_merged_cam_ctx_20_noise_15_atomic_cam_inject_pre_qkv_per_frame_vae",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
},
{
"id": "spatial_mem",
"family": "Spatial",
"paper_row": "Spatial Memory",
"train_recipe": "train/memory_baselines_basic/run_spatial_memory_baseline.sh",
"hf_path": "spatial_mem/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_local/exp1_4_4_cam_rt_paper_style_memory_baselines_basic_spatial_mem",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
},
{
"id": "block_wise_ssm_two_chunk",
"family": "State-space",
"paper_row": "Block-wise SSM",
"train_recipe": "train/memory_baselines_basic/run_ablation_block_wise_ssm_two_chunk.sh",
"hf_path": "block_wise_ssm_two_chunk/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_local/exp1_4_4_cam_rt_paper_style_memory_baselines_basic_abl_block_wise_ssm_two_chunk",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
},
{
"id": "videossm_hybrid",
"family": "State-space",
"paper_row": "Legacy Hybrid (VideoSSM)",
"train_recipe": "train/memory_baselines_basic/run_videossm_hybrid_baseline.sh",
"hf_path": "videossm_hybrid/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_local/exp1_4_4_cam_rt_paper_style_memory_baselines_basic_videossm_hybrid",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
},
{
"id": "spatial_concat_text_two_chunk",
"family": "Spatial",
"paper_row": "Spatial \u2014 concat text readout (ablation)",
"train_recipe": "train/memory_baselines_basic/run_ablation_spatial_concat_text_two_chunk.sh",
"hf_path": "spatial_concat_text_two_chunk/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_local/exp1_4_4_cam_rt_paper_style_memory_baselines_basic_abl_spatial_concat_text_two_chunk",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
},
{
"id": "spatial_inject_none_two_chunk",
"family": "Spatial",
"paper_row": "Spatial \u2014 inject none (ablation)",
"train_recipe": "train/memory_baselines_basic/run_ablation_spatial_inject_none_two_chunk.sh",
"hf_path": "spatial_inject_none_two_chunk/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_local/exp1_4_4_cam_rt_paper_style_memory_baselines_basic_abl_spatial_inject_none_two_chunk",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
},
{
"id": "spatial_cross_attn_readout_t32_g4_two_chunk",
"family": "Spatial",
"paper_row": "Spatial \u2014 cross-attn readout t32 (ablation)",
"train_recipe": "train/memory_baselines_basic/run_ablation_spatial_cross_attn_readout_two_chunk.sh",
"hf_path": "spatial_cross_attn_readout_t32_g4_two_chunk/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_local/exp1_4_4_cam_rt_paper_style_memory_baselines_basic_abl_spatial_cross_attn_readout_t32_g4_two_chunk",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
},
{
"id": "ssm_ablation_ctx1_every4_hint21",
"family": "State-space",
"paper_row": "SSM ablation \u2014 ctx=1, every 4 blocks, hint 21",
"train_recipe": "internal SSM ablation recipe",
"hf_path": "ssm_ablation_ctx1_every4_hint21/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_from_scratch_ssm_ablation_ctx1_every4_hint21",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
},
{
"id": "ssm_ablation_ctx5_every1_hint21",
"family": "State-space",
"paper_row": "SSM ablation \u2014 ctx=5, every 1 block, hint 21",
"train_recipe": "internal SSM ablation recipe",
"hf_path": "ssm_ablation_ctx5_every1_hint21/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_from_scratch_ssm_ablation_ctx5_every1_hint21",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
},
{
"id": "ssm_ablation_ctx5_every4_hint81",
"family": "State-space",
"paper_row": "SSM ablation \u2014 ctx=5, every 4 blocks, hint 81",
"train_recipe": "internal SSM ablation recipe",
"hf_path": "ssm_ablation_ctx5_every4_hint81/epoch-0.safetensors",
"source_subdir": "exp1_4_4_cam_rt_paper_style_from_scratch_ssm_ablation_ctx5_every4_hint81",
"training_steps": 30000,
"backbone": "Wan2.1-T2V-1.3B"
}
]
}