| { |
| "anchor_teacher": "wan_base_motion_teacher", |
| "checkpoint_manifest": { |
| "extra_state": { |
| "best_auroc": 0.9588183889391717, |
| "epoch": 11 |
| }, |
| "files": { |
| "config.json": "535f34a206241cbd1c25afc76f144689cc5ec673ca0fe814dd4642c361871d45", |
| "discriminator.safetensors": "9cdeccb5545f3e6237709dab8be3b78f74c2dc44f72cb13f82c986134266de16", |
| "metrics.json": "ac5416c1effedf6bb37b3a42c833e9278d332d7ceed6bb3e15f092bce2585164", |
| "training_state.pt": "7b5da00a599e76d6ad127092ea83ff8a48fbaab2d0dbc9e7ef225d7c0327fc8f" |
| }, |
| "format": "motion-discriminator-v1", |
| "model_class": "FlowMotionDiscriminator", |
| "model_config": { |
| "blocks_per_stage": 1, |
| "dropout": 0.0, |
| "in_channels": 3, |
| "input_clip": null, |
| "input_scale": 1.0, |
| "mlp_ratio": 4.0, |
| "num_heads": 4, |
| "stem_stride": 4, |
| "transformer_depth": 2, |
| "widths": [ |
| 32, |
| 64, |
| 128 |
| ] |
| }, |
| "step": 19250 |
| }, |
| "checkpoint_name": "checkpoint-19250", |
| "checkpoint_step": 19250, |
| "motion_d_config": { |
| "data": { |
| "clip_frames": 16, |
| "frame_stride": 1, |
| "num_workers": 4, |
| "random_horizontal_flip": 0.0, |
| "random_start": false, |
| "resize_height": 240, |
| "resize_mode": "center_crop", |
| "resize_width": 416, |
| "train_manifest": "motion_phase0_artifacts/manifests/train.jsonl", |
| "val_manifest": "motion_phase0_artifacts/manifests/val.jsonl" |
| }, |
| "evaluation": { |
| "max_abs_magnitude_correlation": 0.7, |
| "min_auroc": 0.8, |
| "min_pairwise_ranking": 0.75 |
| }, |
| "experiment_name": "motion_d_full", |
| "flow": { |
| "append_last_flow": true, |
| "backend": "raft_small", |
| "cache_dir": "motion_phase0_artifacts/flow_cache", |
| "device": "cuda:0", |
| "dtype": "float32", |
| "num_flow_updates": 12, |
| "weights": "motion_phase0_artifacts/weights/raft_small_C_T_V2-01064c6d.pth" |
| }, |
| "hard_negatives": { |
| "enabled": true, |
| "operations": [ |
| { |
| "name": "shuffle", |
| "weight": 1.0 |
| }, |
| { |
| "name": "repeat_drop", |
| "weight": 1.0 |
| }, |
| { |
| "name": "temporal_jitter", |
| "weight": 1.0 |
| }, |
| { |
| "name": "splice", |
| "weight": 1.0 |
| } |
| ], |
| "probability": 1.0 |
| }, |
| "model": { |
| "architecture": "factorized_3d", |
| "blocks_per_stage": 1, |
| "dropout": 0.0, |
| "in_channels": 3, |
| "mlp_ratio": 4.0, |
| "num_heads": 4, |
| "stem_stride": 4, |
| "transformer_depth": 2, |
| "widths": [ |
| 32, |
| 64, |
| 128 |
| ] |
| }, |
| "output": { |
| "directory": "motion_phase0_artifacts/checkpoints/motion_d_full", |
| "keep_last_checkpoints": 3, |
| "overwrite": false |
| }, |
| "schema_version": 1, |
| "seed": 42, |
| "training": { |
| "batch_size": 4, |
| "betas": [ |
| 0.0, |
| 0.99 |
| ], |
| "device": "cuda:0", |
| "epochs": 20, |
| "eval_every_steps": 250, |
| "learning_rate": 0.0002, |
| "log_every_steps": 10, |
| "max_grad_norm": 1.0, |
| "precision": "bf16", |
| "r1_gamma": 1.0, |
| "r2_gamma": 0.0, |
| "regularization_interval": 16, |
| "save_every_steps": 250, |
| "weight_decay": 0.0001 |
| } |
| }, |
| "reward_kind": "logit", |
| "schema_version": 1 |
| } |
|
|
