Upload current M3 Simple step 7365 full resume checkpoint

m3/m3_simple_fulltrain_player_flow_step7365/README.md

@@ -0,0 +1,21 @@
+# M3 Simple — step 7365
+
+Selected checkpoint for continued M3 exploration. Full model and optimizer state are included.
+Code: https://github.com/xixibuxixi23/Plot/tree/efbe000c1c5f8d93a103780b32707f6f90f6b7bb
+This commit packages the implementation after training; it is not the original launch commit.
+
+Architecture: player ROI queries attend player reference images; the resulting spatial
+appearance features are concatenated with noisy video latent, voxel and actor features
+before the joint patch embedder. M3 is fully trainable; Pixel VAE is frozen.
+
+Training: 1 condition + 8 future frames, 8 GPUs × batch 4, no accumulation.
+Loss: full_flow + player_flow + 0.1 player_RGB + 0.025 player_edge.
+Use config.json for the exact architecture, normalization and vocabularies. Paths are
+source-machine provenance: adapt data/output/VAE/index paths on another machine.
+For resume, point --resume at this checkpoint, not config.json's historical parent.
+The original run stopped at step 7365; its configured target 7500 was not reached.
+
+Required separately: matching Pixel VAE and repaired fixed_skins_20260917 data.
+The old identity encoder and old M3 architecture are NOT required.
+See manifest.json for SHA-256, size, lineage and the W&B run.
+This is an experimental resume point, not a final validated player model.

m3/m3_simple_fulltrain_player_flow_step7365/config.json

@@ -0,0 +1,620 @@
+{
+  "renderer": {
+    "num_block_classes": 402,
+    "item_vocab_size": 13,
+    "input_h": 36,
+    "input_w": 64,
+    "in_channels": 16,
+    "hidden_size": 1024,
+    "depth": 12,
+    "num_heads": 16,
+    "voxel_channels": 32,
+    "condition_dim": 256,
+    "actor_channels": 32,
+    "max_agents": 8,
+    "context_frames": 9,
+    "cache_frames": 32,
+    "block_frames": 8,
+    "qk_rms_norm": true,
+    "gradient_checkpointing": true,
+    "gpu_rasterizer": true,
+    "deep_condition_reinjection": false,
+    "view_aware_appearance": false,
+    "detail_preserving_appearance": false,
+    "entity_reference_attention": false,
+    "unified_player_reference": true,
+    "player_reference_grid_size": [
+      4,
+      2
+    ],
+    "player_reference_position_encoding": true,
+    "geometry_aware_player_reference": false,
+    "unified_reference_reinject_blocks": [],
+    "simple_conditioning": true
+  },
+  "codec": {
+    "latent_normalization": {
+      "mode": "per_channel",
+      "mean": [
+        -0.7020555138587952,
+        0.8461390733718872,
+        -0.917613685131073,
+        -1.5030848979949951,
+        -1.698622465133667,
+        -1.8949798345565796,
+        -1.1478651762008667,
+        0.6958867311477661,
+        2.513483762741089,
+        1.2163547277450562,
+        1.9755494594573975,
+        0.7205855250358582,
+        -0.5866942405700684,
+        0.3998744487762451,
+        -2.7082550525665283,
+        -2.014042615890503
+      ],
+      "std": [
+        2.0923426151275635,
+        2.398167848587036,
+        2.862182140350342,
+        1.8666865825653076,
+        2.9419124126434326,
+        2.9730942249298096,
+        2.1236183643341064,
+        2.4726924896240234,
+        2.607072353363037,
+        2.1573493480682373,
+        2.7498817443847656,
+        2.7455546855926514,
+        2.5569992065429688,
+        2.051330804824829,
+        2.2603418827056885,
+        2.367582321166992
+      ],
+      "source": "2daction/stage2_remote_full_47676/latent_stats.npz",
+      "source_sha256": "2746a106b60e8ac20a2e6bd622e226a6db6d0e3aa4d4c79d5a1a1566d454161a",
+      "vae_sha256": "eb634803c94aeea980046961382f2ab67157aa71e92c5a183a35e4f61f8cbc36",
+      "sampled_files": 256,
+      "frames_per_file": 8,
+      "scalar_samples_per_channel": 9437184,
+      "seed": 20260823
+    }
+  },
+  "training": {
+    "dataset_root": "/public/0_DATA/2_Avatar/zhizhou_share/rcz/textagent/data/releases/polis_two_player_fixed_skins_complete_20260917_360p",
+    "train_split": "train",
+    "vocabulary": "derived/common/block_vocabulary.json",
+    "pixel_vae": "checkpoints/pixel_vae/model.safetensors",
+    "latent_normalization": "pixel-vae",
+    "backbone_checkpoint": null,
+    "warm_start": null,
+    "output_dir": "/root/rcz-runs/m3_fulltrain_8gpu_b4_player_flow_20260919",
+    "window_index": "/public/0_DATA/2_Avatar/zhizhou_share/rcz/Plot/derived/m3_fulltrain_c9_20260919/train_c9.pt",
+    "chunk_cache_root": null,
+    "appearance_counterfactual": false,
+    "counterfactual_variants": 4,
+    "counterfactual_dataset_root": null,
+    "counterfactual_window_index": null,
+    "counterfactual_chunk_cache_root": null,
+    "counterfactual_step_probability": 0.0,
+    "counterfactual_player_mask_probability": 1.0,
+    "counterfactual_random_timesteps": false,
+    "val_window_index": "/public/0_DATA/2_Avatar/zhizhou_share/rcz/Plot/derived/m3_player_short_20260919/val_id_c9.pt",
+    "health_focus_index": null,
+    "health_focus_oversample": 1,
+    "resume": "/root/rcz-runs/m3_fulltrain_player_flow_from6500_c9_20260919/step_0006744.pt",
+    "allow_item_vocabulary_extension": false,
+    "deep_condition_reinjection": false,
+    "simple_m3": true,
+    "view_aware_appearance": false,
+    "detail_preserving_appearance": false,
+    "entity_reference_attention": false,
+    "unified_player_reference": false,
+    "unified_reference_reinject_blocks": [],
+    "player_reference_token_grid": [
+      4,
+      2
+    ],
+    "player_reference_position_encoding": true,
+    "geometry_aware_player_reference": false,
+    "freeze_base_for_appearance": false,
+    "freeze_base_for_reference": false,
+    "freeze_base_for_player_appearance": false,
+    "appearance_unfreeze_last_spatial_blocks": 0,
+    "reference_unfreeze_last_spatial_blocks": 0,
+    "context_frames": 9,
+    "cache_frames": 32,
+    "block_frames": 8,
+    "qk_rms_norm": true,
+    "hidden_size": 1024,
+    "depth": 12,
+    "heads": 16,
+    "voxel_channels": 32,
+    "condition_dim": 256,
+    "actor_channels": 32,
+    "target_views_per_window": 1,
+    "batch_size": 4,
+    "workers": 2,
+    "prefetch_factor": 2,
+    "steps": 7500,
+    "save_every": 250,
+    "validate_every": 250,
+    "visualize_every": 250,
+    "visualization_denoising_steps": 20,
+    "visualization_probe_manifest": "/public/0_DATA/2_Avatar/zhizhou_share/rcz/Plot/derived/m3_player_short_20260919/probes.json",
+    "val_batches": 8,
+    "log_every": 10,
+    "gradient_accumulation": 1,
+    "precision": "bf16",
+    "loss_mode": "combined",
+    "flow_loss_weight": 1.0,
+    "player_flow_loss_weight": 1.0,
+    "profile_local_step": 0,
+    "latent_entity_region_upweight": 0.0,
+    "latent_player_region_upweight": 0.0,
+    "player_mask_probability": 0.5,
+    "pixel_loss_frames": 2,
+    "pixel_frame_selection": "player",
+    "entity_pixel_l1_weight": 0.0,
+    "entity_pixel_edge_weight": 0.0,
+    "player_pixel_l1_weight": 0.1,
+    "player_pixel_edge_weight": 0.025,
+    "health_pixel_l1_weight": 0.0,
+    "damaged_health_upweight": 4.0,
+    "player_identity_checkpoint": null,
+    "player_identity_loss_weight": 0.0,
+    "player_identity_margin": 0.2,
+    "player_identity_negative_weight": 0.5,
+    "counterfactual_player_difference_weight": 0.0,
+    "mask_prefix_player_probability": 0.0,
+    "counterfactual_mask_prefix_player_probability": 1.0,
+    "lr": 1e-05,
+    "appearance_lr": 5e-05,
+    "unfrozen_base_lr_scale": 1.0,
+    "seed": 20260919,
+    "device": "cuda",
+    "wandb_entity": "ckx23-tsinghua-university",
+    "wandb_project": "plot-m3",
+    "wandb_name": "m3-fulltrain-c9-player-flow-8gpu-b4-from6744",
+    "wandb_mode": "online",
+    "checkpoint_staging_dir": "/root/rcz-runs/m3_fulltrain_8gpu_b4_player_flow_20260919/staging",
+    "checkpoint_errors": "raise"
+  },
+  "item_vocabulary": {
+    "mcl_core:brick_block": 1,
+    "mcl_core:cobble": 2,
+    "mcl_core:dirt": 3,
+    "mcl_core:glass": 4,
+    "mcl_core:stonebrick": 5,
+    "mcl_core:tree": 6,
+    "mcl_core:wood": 7,
+    "mcl_ocean:sea_lantern": 8,
+    "mcl_tools:pick_iron": 9,
+    "textagent_task:npc_axe": 10,
+    "textagent_task:npc_sword": 11,
+    "mcl_tools:shovel_iron": 12
+  },
+  "class_to_raw": [
+    1,
+    9,
+    13,
+    15,
+    25,
+    26,
+    31,
+    51,
+    52,
+    54,
+    56,
+    57,
+    59,
+    60,
+    61,
+    63,
+    64,
+    65,
+    87,
+    88,
+    89,
+    91,
+    92,
+    93,
+    95,
+    96,
+    97,
+    120,
+    126,
+    156,
+    157,
+    158,
+    160,
+    161,
+    162,
+    164,
+    165,
+    166,
+    188,
+    191,
+    192,
+    193,
+    195,
+    201,
+    203,
+    216,
+    217,
+    218,
+    219,
+    220,
+    223,
+    226,
+    227,
+    228,
+    229,
+    231,
+    232,
+    233,
+    235,
+    237,
+    238,
+    239,
+    240,
+    242,
+    244,
+    245,
+    246,
+    248,
+    249,
+    250,
+    251,
+    252,
+    253,
+    254,
+    258,
+    259,
+    261,
+    262,
+    266,
+    267,
+    270,
+    271,
+    272,
+    273,
+    278,
+    286,
+    287,
+    288,
+    289,
+    290,
+    291,
+    292,
+    293,
+    295,
+    297,
+    299,
+    301,
+    303,
+    305,
+    309,
+    313,
+    314,
+    317,
+    318,
+    319,
+    320,
+    322,
+    329,
+    330,
+    331,
+    332,
+    333,
+    334,
+    335,
+    336,
+    337,
+    338,
+    339,
+    341,
+    358,
+    359,
+    360,
+    362,
+    363,
+    382,
+    384,
+    394,
+    418,
+    421,
+    423,
+    424,
+    429,
+    434,
+    435,
+    437,
+    438,
+    450,
+    453,
+    456,
+    462,
+    464,
+    468,
+    474,
+    476,
+    477,
+    486,
+    487,
+    488,
+    489,
+    490,
+    491,
+    493,
+    494,
+    495,
+    496,
+    497,
+    499,
+    500,
+    501,
+    502,
+    503,
+    505,
+    506,
+    507,
+    508,
+    509,
+    511,
+    512,
+    513,
+    514,
+    515,
+    517,
+    518,
+    519,
+    520,
+    521,
+    523,
+    524,
+    525,
+    526,
+    528,
+    530,
+    532,
+    533,
+    534,
+    536,
+    566,
+    568,
+    571,
+    574,
+    577,
+    717,
+    720,
+    723,
+    726,
+    729,
+    755,
+    758,
+    759,
+    850,
+    852,
+    853,
+    854,
+    868,
+    869,
+    871,
+    872,
+    878,
+    880,
+    883,
+    884,
+    885,
+    892,
+    893,
+    894,
+    913,
+    914,
+    915,
+    916,
+    919,
+    931,
+    932,
+    934,
+    955,
+    956,
+    973,
+    974,
+    976,
+    1003,
+    1004,
+    1012,
+    1039,
+    1045,
+    1116,
+    1117,
+    1118,
+    1122,
+    1129,
+    1154,
+    1161,
+    1170,
+    1171,
+    1181,
+    1191,
+    1196,
+    1221,
+    1229,
+    1231,
+    1281,
+    1285,
+    1296,
+    1305,
+    1409,
+    1411,
+    1412,
+    1414,
+    1431,
+    1432,
+    1464,
+    1465,
+    1467,
+    1471,
+    1472,
+    1473,
+    1474,
+    1495,
+    1496,
+    1513,
+    1514,
+    1527,
+    1528,
+    1536,
+    1537,
+    1540,
+    1543,
+    1546,
+    1549,
+    1551,
+    1553,
+    1556,
+    1557,
+    1558,
+    1560,
+    1601,
+    1607,
+    1619,
+    1764,
+    1765,
+    1770,
+    1778,
+    1796,
+    1802,
+    1803,
+    1844,
+    1850,
+    1859,
+    1915,
+    1942,
+    1946,
+    1947,
+    1948,
+    1950,
+    1951,
+    1995,
+    2030,
+    2031,
+    2032,
+    2034,
+    2035,
+    2037,
+    2038,
+    2040,
+    2041,
+    2045,
+    2053,
+    2062,
+    2066,
+    2077,
+    2081,
+    2090,
+    2098,
+    2106,
+    2107,
+    2108,
+    2119,
+    2122,
+    2126,
+    2128,
+    2134,
+    2136,
+    2157,
+    2158,
+    2159,
+    2160,
+    2161,
+    2162,
+    2163,
+    2164,
+    2165,
+    2166,
+    2167,
+    2168,
+    2169,
+    2170,
+    2171,
+    2172,
+    2173,
+    2174,
+    2175,
+    2176,
+    2177,
+    2178,
+    2179,
+    2180,
+    2181,
+    2182,
+    2183,
+    2185,
+    2187,
+    2188,
+    2189,
+    2190,
+    2191,
+    2192,
+    2193,
+    2194,
+    2197,
+    2199,
+    2201,
+    2209,
+    2212,
+    2213,
+    2214,
+    2221,
+    2273,
+    2274,
+    2280,
+    2281,
+    2282,
+    2283,
+    2284,
+    2285,
+    2286,
+    2287,
+    2288,
+    2289,
+    2290,
+    2291,
+    2292,
+    2293,
+    2294,
+    2328,
+    2329,
+    2346,
+    2350,
+    2416,
+    2420,
+    2424,
+    2428,
+    2432,
+    2436,
+    2456
+  ],
+  "optimizer_groups": [
+    {
+      "name": "backbone_and_conditions",
+      "lr": 1e-05,
+      "parameters": 457241554
+    },
+    {
+      "name": "appearance",
+      "lr": 5e-05,
+      "parameters": 415080
+    },
+    {
+      "name": "joint_patch",
+      "lr": 1e-05,
+      "parameters": 3407872,
+      "appearance_lr": 5e-05,
+      "appearance_input_channels": 32
+    }
+  ]
+}

m3/m3_simple_fulltrain_player_flow_step7365/manifest.json

@@ -0,0 +1,55 @@
+{
+  "checkpoint": {
+    "path": "m3/m3_simple_fulltrain_player_flow_step7365/step_0007365.pt",
+    "role": "selected current M3 exploration/resume checkpoint",
+    "bytes": 5533183645,
+    "sha256": "b598bfc041177d152841484df5e9e0ef5cf17594c70031d24316bf01b1aef80c",
+    "step": 7365,
+    "source": "m3_fulltrain_8gpu_b4_player_flow_20260919",
+    "architecture": "M3 Simple: ROI appearance attention then channel concat with video/voxel/actor maps into joint patch embedder",
+    "compatible_code_commit": "efbe000c1c5f8d93a103780b32707f6f90f6b7bb",
+    "code_note": "This is the subsequent committed implementation, not a claim that the training launched at this commit.",
+    "dataset_release": "polis_two_player_fixed_skins_complete_20260917_360p (repaired references)",
+    "contains": [
+      "model",
+      "optimizer"
+    ],
+    "wandb": "https://wandb.ai/ckx23-tsinghua-university/plot-m3/runs/fhh63zsg",
+    "loss": "full_flow + player_flow + 0.1 * player_RGB + 0.025 * player_edge",
+    "batch": {
+      "per_gpu": 4,
+      "gpus": 8,
+      "gradient_accumulation": 1,
+      "effective": 32
+    },
+    "context_frames": 9,
+    "condition_frames": 1,
+    "rollout_block_frames": 8,
+    "dependencies": {
+      "pixel_vae_sha256": "eb634803c94aeea980046961382f2ab67157aa71e92c5a183a35e4f61f8cbc36"
+    },
+    "limitations": "Exploration checkpoint; not claimed to solve player appearance. Validation logs use a small fixed probe set."
+  },
+  "files": {
+    "step_0007365.pt": {
+      "bytes": 5533183645,
+      "sha256": "b598bfc041177d152841484df5e9e0ef5cf17594c70031d24316bf01b1aef80c"
+    },
+    "config.json": {
+      "bytes": 10751,
+      "sha256": "683cac48e0091e21d44662c0e594afcb43e51391d0fed8583a1d1117ba40575f"
+    },
+    "training.jsonl": {
+      "bytes": 57666,
+      "sha256": "e96e4f9b1e9e1f116cfa380d3797b5a9ffc11d9993a3f805e5638b126156b779"
+    },
+    "validation.jsonl": {
+      "bytes": 1562,
+      "sha256": "3f4d407fffb37a45ed10c8d5a152c51379ea3796843c4d64a0101cd2d2c8eed5"
+    },
+    "wandb_run.json": {
+      "bytes": 164,
+      "sha256": "bae95603af07308525c2b5953a60134141f42399c02d839ce006d8fc70c2c4c4"
+    }
+  }
+}

m3/m3_simple_fulltrain_player_flow_step7365/step_0007365.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b598bfc041177d152841484df5e9e0ef5cf17594c70031d24316bf01b1aef80c
+size 5533183645

m3/m3_simple_fulltrain_player_flow_step7365/training.jsonl

@@ -0,0 +1,62 @@
+{"step": 6750, "train/total_loss": 0.6943467855453491, "train/flow_loss": 0.19159415364265442, "train/player_flow_loss": 0.5000977516174316, "train/auxiliary_loss": 0.002654965966939926, "train/entity_pixel_l1": 0.01795738935470581, "train/entity_pixel_edge": 0.010307122953236103, "train/player_pixel_l1": 0.023264501243829727, "train/player_pixel_edge": 0.013140637427568436, "train/health_pixel_l1": 0.03857408091425896, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.705316543579102, "train/cuda_peak_allocated_max_gib": 20.002941131591797, "train/cuda_peak_reserved_max_gib": 21.2890625, "train/counterfactual_step": 0.0}
+{"step": 6760, "train/total_loss": 0.6818394660949707, "train/flow_loss": 0.23693718016147614, "train/player_flow_loss": 0.4412102699279785, "train/auxiliary_loss": 0.0036920879501849413, "train/entity_pixel_l1": 0.02555812895298004, "train/entity_pixel_edge": 0.012124781496822834, "train/player_pixel_l1": 0.03290536254644394, "train/player_pixel_edge": 0.01606205850839615, "train/health_pixel_l1": 0.03961366415023804, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723308563232422, "train/cuda_peak_allocated_max_gib": 20.015531539916992, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6770, "train/total_loss": 0.7725566029548645, "train/flow_loss": 0.205086350440979, "train/player_flow_loss": 0.5625168085098267, "train/auxiliary_loss": 0.004953403491526842, "train/entity_pixel_l1": 0.028422759845852852, "train/entity_pixel_edge": 0.012692073360085487, "train/player_pixel_l1": 0.045000456273555756, "train/player_pixel_edge": 0.01813431829214096, "train/health_pixel_l1": 0.043128322809934616, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723061561584473, "train/cuda_peak_allocated_max_gib": 20.01762580871582, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6780, "train/total_loss": 0.828368604183197, "train/flow_loss": 0.26266732811927795, "train/player_flow_loss": 0.5619328618049622, "train/auxiliary_loss": 0.0037684724666178226, "train/entity_pixel_l1": 0.027457818388938904, "train/entity_pixel_edge": 0.015346363186836243, "train/player_pixel_l1": 0.0328943207859993, "train/player_pixel_edge": 0.019161611795425415, "train/health_pixel_l1": 0.04123424366116524, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72364616394043, "train/cuda_peak_allocated_max_gib": 20.01762580871582, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6790, "train/total_loss": 0.8581790924072266, "train/flow_loss": 0.2279176414012909, "train/player_flow_loss": 0.6253631711006165, "train/auxiliary_loss": 0.0048982733860611916, "train/entity_pixel_l1": 0.024093713611364365, "train/entity_pixel_edge": 0.014438563957810402, "train/player_pixel_l1": 0.04348267614841461, "train/player_pixel_edge": 0.022000238299369812, "train/health_pixel_l1": 0.03882724791765213, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72364616394043, "train/cuda_peak_allocated_max_gib": 20.020936965942383, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6800, "train/total_loss": 0.8254378437995911, "train/flow_loss": 0.25456660985946655, "train/player_flow_loss": 0.5661582350730896, "train/auxiliary_loss": 0.004712940193712711, "train/entity_pixel_l1": 0.026341885328292847, "train/entity_pixel_edge": 0.012859629467129707, "train/player_pixel_l1": 0.04240996390581131, "train/player_pixel_edge": 0.018877742812037468, "train/health_pixel_l1": 0.03871859610080719, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724687576293945, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6810, "train/total_loss": 0.6504104733467102, "train/flow_loss": 0.2173978090286255, "train/player_flow_loss": 0.42966365814208984, "train/auxiliary_loss": 0.0033490448258817196, "train/entity_pixel_l1": 0.023020867258310318, "train/entity_pixel_edge": 0.01253130566328764, "train/player_pixel_l1": 0.03016422688961029, "train/player_pixel_edge": 0.01330488733947277, "train/health_pixel_l1": 0.039573729038238525, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.71690559387207, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6820, "train/total_loss": 0.737209141254425, "train/flow_loss": 0.25011977553367615, "train/player_flow_loss": 0.48305875062942505, "train/auxiliary_loss": 0.004030629992485046, "train/entity_pixel_l1": 0.020848441869020462, "train/entity_pixel_edge": 0.011290542781352997, "train/player_pixel_l1": 0.035548023879528046, "train/player_pixel_edge": 0.01903311349451542, "train/health_pixel_l1": 0.04201311618089676, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724199295043945, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6830, "train/total_loss": 0.6182664036750793, "train/flow_loss": 0.20329056680202484, "train/player_flow_loss": 0.41076284646987915, "train/auxiliary_loss": 0.004212950821965933, "train/entity_pixel_l1": 0.02632334642112255, "train/entity_pixel_edge": 0.013377899304032326, "train/player_pixel_l1": 0.037453021854162216, "train/player_pixel_edge": 0.01870594173669815, "train/health_pixel_l1": 0.03808177635073662, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72291374206543, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6840, "train/total_loss": 0.6570245027542114, "train/flow_loss": 0.22446765005588531, "train/player_flow_loss": 0.4286568760871887, "train/auxiliary_loss": 0.0038999528624117374, "train/entity_pixel_l1": 0.020695233717560768, "train/entity_pixel_edge": 0.012630289420485497, "train/player_pixel_l1": 0.03495696187019348, "train/player_pixel_edge": 0.016170261427760124, "train/health_pixel_l1": 0.03829409182071686, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723001480102539, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6850, "train/total_loss": 0.7764411568641663, "train/flow_loss": 0.33522844314575195, "train/player_flow_loss": 0.4375673532485962, "train/auxiliary_loss": 0.00364536140114069, "train/entity_pixel_l1": 0.023708289489150047, "train/entity_pixel_edge": 0.012581893242895603, "train/player_pixel_l1": 0.0321548730134964, "train/player_pixel_edge": 0.017194965854287148, "train/health_pixel_l1": 0.040112122893333435, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723308563232422, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6860, "train/total_loss": 0.6793588399887085, "train/flow_loss": 0.20902854204177856, "train/player_flow_loss": 0.4674457907676697, "train/auxiliary_loss": 0.002884489018470049, "train/entity_pixel_l1": 0.02189648523926735, "train/entity_pixel_edge": 0.011207561939954758, "train/player_pixel_l1": 0.025547336786985397, "train/player_pixel_edge": 0.013190213590860367, "train/health_pixel_l1": 0.03618885949254036, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723268508911133, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6870, "train/total_loss": 0.6898401975631714, "train/flow_loss": 0.21448470652103424, "train/player_flow_loss": 0.4710366725921631, "train/auxiliary_loss": 0.0043188175186514854, "train/entity_pixel_l1": 0.02579944208264351, "train/entity_pixel_edge": 0.012072362937033176, "train/player_pixel_l1": 0.03864102065563202, "train/player_pixel_edge": 0.018188612535595894, "train/health_pixel_l1": 0.04234690219163895, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72315788269043, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6880, "train/total_loss": 0.6796979308128357, "train/flow_loss": 0.21931341290473938, "train/player_flow_loss": 0.4555143415927887, "train/auxiliary_loss": 0.004870148375630379, "train/entity_pixel_l1": 0.029836179688572884, "train/entity_pixel_edge": 0.012661758810281754, "train/player_pixel_l1": 0.04456302896142006, "train/player_pixel_edge": 0.016553789377212524, "train/health_pixel_l1": 0.049815114587545395, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723920822143555, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6890, "train/total_loss": 0.7602676153182983, "train/flow_loss": 0.22159627079963684, "train/player_flow_loss": 0.534127414226532, "train/auxiliary_loss": 0.004543991759419441, "train/entity_pixel_l1": 0.020731734111905098, "train/entity_pixel_edge": 0.01306739542633295, "train/player_pixel_l1": 0.03940309211611748, "train/player_pixel_edge": 0.024147287011146545, "train/health_pixel_l1": 0.036646004766225815, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72307014465332, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6900, "train/total_loss": 0.773186981678009, "train/flow_loss": 0.17553487420082092, "train/player_flow_loss": 0.5941883325576782, "train/auxiliary_loss": 0.003463851986452937, "train/entity_pixel_l1": 0.018111055716872215, "train/entity_pixel_edge": 0.010297750122845173, "train/player_pixel_l1": 0.03051169402897358, "train/player_pixel_edge": 0.016507305204868317, "train/health_pixel_l1": 0.036726366728544235, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723308563232422, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6910, "train/total_loss": 0.7469326257705688, "train/flow_loss": 0.17558416724205017, "train/player_flow_loss": 0.5675881505012512, "train/auxiliary_loss": 0.0037602989468723536, "train/entity_pixel_l1": 0.021476222202181816, "train/entity_pixel_edge": 0.011600162833929062, "train/player_pixel_l1": 0.03323615342378616, "train/player_pixel_edge": 0.017467336729168892, "train/health_pixel_l1": 0.04021058976650238, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723169326782227, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6920, "train/total_loss": 0.6177505254745483, "train/flow_loss": 0.20230498909950256, "train/player_flow_loss": 0.411793053150177, "train/auxiliary_loss": 0.0036525155883282423, "train/entity_pixel_l1": 0.020012473687529564, "train/entity_pixel_edge": 0.01165053527802229, "train/player_pixel_l1": 0.03223007917404175, "train/player_pixel_edge": 0.017180297523736954, "train/health_pixel_l1": 0.04205601289868355, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723308563232422, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6930, "train/total_loss": 0.6752729415893555, "train/flow_loss": 0.21640199422836304, "train/player_flow_loss": 0.4553987681865692, "train/auxiliary_loss": 0.003472232958301902, "train/entity_pixel_l1": 0.02113838866353035, "train/entity_pixel_edge": 0.012587394565343857, "train/player_pixel_l1": 0.030597317963838577, "train/player_pixel_edge": 0.01650002971291542, "train/health_pixel_l1": 0.03876882791519165, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.722322463989258, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6940, "train/total_loss": 0.7675078511238098, "train/flow_loss": 0.26382771134376526, "train/player_flow_loss": 0.49940380454063416, "train/auxiliary_loss": 0.004276289604604244, "train/entity_pixel_l1": 0.022218842059373856, "train/entity_pixel_edge": 0.012398901395499706, "train/player_pixel_l1": 0.038502249866724014, "train/player_pixel_edge": 0.017042560502886772, "train/health_pixel_l1": 0.041470710188150406, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724409103393555, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6950, "train/total_loss": 0.5471312999725342, "train/flow_loss": 0.14528019726276398, "train/player_flow_loss": 0.39726951718330383, "train/auxiliary_loss": 0.004581596236675978, "train/entity_pixel_l1": 0.02170654758810997, "train/entity_pixel_edge": 0.011719542555510998, "train/player_pixel_l1": 0.041641753166913986, "train/player_pixel_edge": 0.016696840524673462, "train/health_pixel_l1": 0.03887026011943817, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724409103393555, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6960, "train/total_loss": 0.6809165477752686, "train/flow_loss": 0.2014465481042862, "train/player_flow_loss": 0.47488534450531006, "train/auxiliary_loss": 0.004584658425301313, "train/entity_pixel_l1": 0.026977986097335815, "train/entity_pixel_edge": 0.012276259250938892, "train/player_pixel_l1": 0.04145307093858719, "train/player_pixel_edge": 0.01757405325770378, "train/health_pixel_l1": 0.035641372203826904, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723064422607422, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6970, "train/total_loss": 0.6714326739311218, "train/flow_loss": 0.17020626366138458, "train/player_flow_loss": 0.49620765447616577, "train/auxiliary_loss": 0.005018802359700203, "train/entity_pixel_l1": 0.023061569780111313, "train/entity_pixel_edge": 0.012792997062206268, "train/player_pixel_l1": 0.04489465057849884, "train/player_pixel_edge": 0.021173477172851562, "train/health_pixel_l1": 0.04061330109834671, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723549842834473, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6980, "train/total_loss": 0.7507927417755127, "train/flow_loss": 0.24672505259513855, "train/player_flow_loss": 0.49982181191444397, "train/auxiliary_loss": 0.004245922900736332, "train/entity_pixel_l1": 0.02353139966726303, "train/entity_pixel_edge": 0.01266729086637497, "train/player_pixel_l1": 0.03814272955060005, "train/player_pixel_edge": 0.01726599410176277, "train/health_pixel_l1": 0.03567630797624588, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72413444519043, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 6990, "train/total_loss": 0.855323314666748, "train/flow_loss": 0.26357486844062805, "train/player_flow_loss": 0.5875340104103088, "train/auxiliary_loss": 0.004214515443891287, "train/entity_pixel_l1": 0.0252920500934124, "train/entity_pixel_edge": 0.012211065739393234, "train/player_pixel_l1": 0.03838852047920227, "train/player_pixel_edge": 0.01502652745693922, "train/health_pixel_l1": 0.03748456388711929, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723308563232422, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7000, "train/total_loss": 0.6503363847732544, "train/flow_loss": 0.17432613670825958, "train/player_flow_loss": 0.4720144271850586, "train/auxiliary_loss": 0.003995782695710659, "train/entity_pixel_l1": 0.02274136245250702, "train/entity_pixel_edge": 0.01205477025359869, "train/player_pixel_l1": 0.03551905229687691, "train/player_pixel_edge": 0.017755111679434776, "train/health_pixel_l1": 0.040937550365924835, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723308563232422, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7010, "train/total_loss": 0.7881531119346619, "train/flow_loss": 0.22731076180934906, "train/player_flow_loss": 0.557369589805603, "train/auxiliary_loss": 0.003472850425168872, "train/entity_pixel_l1": 0.023137442767620087, "train/entity_pixel_edge": 0.010784119367599487, "train/player_pixel_l1": 0.031233880668878555, "train/player_pixel_edge": 0.013978485018014908, "train/health_pixel_l1": 0.04102097451686859, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723112106323242, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7020, "train/total_loss": 0.7443345785140991, "train/flow_loss": 0.23961308598518372, "train/player_flow_loss": 0.49955639243125916, "train/auxiliary_loss": 0.005165104754269123, "train/entity_pixel_l1": 0.030367985367774963, "train/entity_pixel_edge": 0.014457921497523785, "train/player_pixel_l1": 0.0468331053853035, "train/player_pixel_edge": 0.01927177421748638, "train/health_pixel_l1": 0.04282881319522858, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723390579223633, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7030, "train/total_loss": 0.7167684435844421, "train/flow_loss": 0.2317425012588501, "train/player_flow_loss": 0.4808449149131775, "train/auxiliary_loss": 0.0041810572147369385, "train/entity_pixel_l1": 0.025653861463069916, "train/entity_pixel_edge": 0.010931747034192085, "train/player_pixel_l1": 0.0383356437087059, "train/player_pixel_edge": 0.013899723067879677, "train/health_pixel_l1": 0.040546126663684845, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723356246948242, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7040, "train/total_loss": 0.7211030721664429, "train/flow_loss": 0.22173455357551575, "train/player_flow_loss": 0.4951106905937195, "train/auxiliary_loss": 0.004257826134562492, "train/entity_pixel_l1": 0.023242857307195663, "train/entity_pixel_edge": 0.012790856882929802, "train/player_pixel_l1": 0.037498313933610916, "train/player_pixel_edge": 0.02031976915895939, "train/health_pixel_l1": 0.03934553265571594, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724393844604492, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7050, "train/total_loss": 0.5587999224662781, "train/flow_loss": 0.18880042433738708, "train/player_flow_loss": 0.36584529280662537, "train/auxiliary_loss": 0.00415418017655611, "train/entity_pixel_l1": 0.021879183128476143, "train/entity_pixel_edge": 0.012450266629457474, "train/player_pixel_l1": 0.03680609539151192, "train/player_pixel_edge": 0.01894282177090645, "train/health_pixel_l1": 0.03723117336630821, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724409103393555, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7060, "train/total_loss": 0.5773053169250488, "train/flow_loss": 0.16766613721847534, "train/player_flow_loss": 0.40582481026649475, "train/auxiliary_loss": 0.0038143754936754704, "train/entity_pixel_l1": 0.020477350801229477, "train/entity_pixel_edge": 0.012180492281913757, "train/player_pixel_l1": 0.03311637416481972, "train/player_pixel_edge": 0.02010953053832054, "train/health_pixel_l1": 0.0384979210793972, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.721355438232422, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7070, "train/total_loss": 0.8070588111877441, "train/flow_loss": 0.2419622540473938, "train/player_flow_loss": 0.5608384609222412, "train/auxiliary_loss": 0.004258064553141594, "train/entity_pixel_l1": 0.028861695900559425, "train/entity_pixel_edge": 0.012926485389471054, "train/player_pixel_l1": 0.03859306871891022, "train/player_pixel_edge": 0.015950307250022888, "train/health_pixel_l1": 0.04186875745654106, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72389030456543, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7080, "train/total_loss": 0.7350791096687317, "train/flow_loss": 0.21949328482151031, "train/player_flow_loss": 0.5116105079650879, "train/auxiliary_loss": 0.00397527776658535, "train/entity_pixel_l1": 0.02384892851114273, "train/entity_pixel_edge": 0.012783216312527657, "train/player_pixel_l1": 0.03532468527555466, "train/player_pixel_edge": 0.017712371423840523, "train/health_pixel_l1": 0.04176473617553711, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72370719909668, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7090, "train/total_loss": 0.6000446081161499, "train/flow_loss": 0.184036985039711, "train/player_flow_loss": 0.4116087555885315, "train/auxiliary_loss": 0.0043988535180687904, "train/entity_pixel_l1": 0.02300046570599079, "train/entity_pixel_edge": 0.012294032610952854, "train/player_pixel_l1": 0.03947696089744568, "train/player_pixel_edge": 0.018046291545033455, "train/health_pixel_l1": 0.03621857985854149, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72477912902832, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7100, "train/total_loss": 0.825819194316864, "train/flow_loss": 0.2527301609516144, "train/player_flow_loss": 0.568440854549408, "train/auxiliary_loss": 0.004648165311664343, "train/entity_pixel_l1": 0.023823123425245285, "train/entity_pixel_edge": 0.01258446928113699, "train/player_pixel_l1": 0.04174624755978584, "train/player_pixel_edge": 0.018941618502140045, "train/health_pixel_l1": 0.04666983708739281, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723356246948242, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7110, "train/total_loss": 0.621709406375885, "train/flow_loss": 0.17501585185527802, "train/player_flow_loss": 0.44347110390663147, "train/auxiliary_loss": 0.003222440369427204, "train/entity_pixel_l1": 0.019110940396785736, "train/entity_pixel_edge": 0.011569146066904068, "train/player_pixel_l1": 0.028067782521247864, "train/player_pixel_edge": 0.016626490280032158, "train/health_pixel_l1": 0.03765324875712395, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724897384643555, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7120, "train/total_loss": 0.682144284248352, "train/flow_loss": 0.18903256952762604, "train/player_flow_loss": 0.4887310862541199, "train/auxiliary_loss": 0.004380614496767521, "train/entity_pixel_l1": 0.024602007120847702, "train/entity_pixel_edge": 0.013225341215729713, "train/player_pixel_l1": 0.03884705901145935, "train/player_pixel_edge": 0.01983632519841194, "train/health_pixel_l1": 0.03855849429965019, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724897384643555, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7130, "train/total_loss": 0.5729793906211853, "train/flow_loss": 0.16926297545433044, "train/player_flow_loss": 0.3996890187263489, "train/auxiliary_loss": 0.004027487710118294, "train/entity_pixel_l1": 0.024752814322710037, "train/entity_pixel_edge": 0.012926895171403885, "train/player_pixel_l1": 0.03578764945268631, "train/player_pixel_edge": 0.01794891618192196, "train/health_pixel_l1": 0.04116085171699524, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72315788269043, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.3984375, "train/counterfactual_step": 0.0}
+{"step": 7140, "train/total_loss": 0.7009141445159912, "train/flow_loss": 0.2354683130979538, "train/player_flow_loss": 0.46052950620651245, "train/auxiliary_loss": 0.004916345234960318, "train/entity_pixel_l1": 0.027705805376172066, "train/entity_pixel_edge": 0.012805336154997349, "train/player_pixel_l1": 0.044658761471509933, "train/player_pixel_edge": 0.01801874488592148, "train/health_pixel_l1": 0.03809622675180435, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.718502044677734, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7150, "train/total_loss": 0.8128835558891296, "train/flow_loss": 0.245370015501976, "train/player_flow_loss": 0.5632515549659729, "train/auxiliary_loss": 0.004262012429535389, "train/entity_pixel_l1": 0.025097260251641273, "train/entity_pixel_edge": 0.012134669348597527, "train/player_pixel_l1": 0.038277942687273026, "train/player_pixel_edge": 0.017368733882904053, "train/health_pixel_l1": 0.04116878658533096, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723356246948242, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7160, "train/total_loss": 0.6477146744728088, "train/flow_loss": 0.173920676112175, "train/player_flow_loss": 0.46864378452301025, "train/auxiliary_loss": 0.005150268319994211, "train/entity_pixel_l1": 0.0320005789399147, "train/entity_pixel_edge": 0.012758101336658001, "train/player_pixel_l1": 0.04727870225906372, "train/player_pixel_edge": 0.01689591258764267, "train/health_pixel_l1": 0.03693452477455139, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72389030456543, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7170, "train/total_loss": 0.6032591462135315, "train/flow_loss": 0.19999858736991882, "train/player_flow_loss": 0.3990105986595154, "train/auxiliary_loss": 0.004249920137226582, "train/entity_pixel_l1": 0.02799852006137371, "train/entity_pixel_edge": 0.012791302055120468, "train/player_pixel_l1": 0.03851901367306709, "train/player_pixel_edge": 0.015920741483569145, "train/health_pixel_l1": 0.03927389532327652, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.717401504516602, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7180, "train/total_loss": 0.6696106791496277, "train/flow_loss": 0.20725736021995544, "train/player_flow_loss": 0.45707833766937256, "train/auxiliary_loss": 0.00527497474104166, "train/entity_pixel_l1": 0.0281973909586668, "train/entity_pixel_edge": 0.011785909533500671, "train/player_pixel_l1": 0.048837870359420776, "train/player_pixel_edge": 0.01564750261604786, "train/health_pixel_l1": 0.03822225704789162, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.720714569091797, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7190, "train/total_loss": 0.783330500125885, "train/flow_loss": 0.23697204887866974, "train/player_flow_loss": 0.5419302582740784, "train/auxiliary_loss": 0.004428122658282518, "train/entity_pixel_l1": 0.024184122681617737, "train/entity_pixel_edge": 0.013710463419556618, "train/player_pixel_l1": 0.03916095569729805, "train/player_pixel_edge": 0.020481087267398834, "train/health_pixel_l1": 0.040664251893758774, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723920822143555, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7200, "train/total_loss": 0.6466741561889648, "train/flow_loss": 0.22065554559230804, "train/player_flow_loss": 0.4222494661808014, "train/auxiliary_loss": 0.0037690913304686546, "train/entity_pixel_l1": 0.021568335592746735, "train/entity_pixel_edge": 0.01278148777782917, "train/player_pixel_l1": 0.03321392089128494, "train/player_pixel_edge": 0.017907967790961266, "train/health_pixel_l1": 0.0434039905667305, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72364616394043, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7210, "train/total_loss": 0.6674860119819641, "train/flow_loss": 0.20003356039524078, "train/player_flow_loss": 0.4628978371620178, "train/auxiliary_loss": 0.004554652143269777, "train/entity_pixel_l1": 0.028654757887125015, "train/entity_pixel_edge": 0.0136162880808115, "train/player_pixel_l1": 0.041141293942928314, "train/player_pixel_edge": 0.017620902508497238, "train/health_pixel_l1": 0.04174332693219185, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723061561584473, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7220, "train/total_loss": 0.6952368021011353, "train/flow_loss": 0.24682900309562683, "train/player_flow_loss": 0.44414180517196655, "train/auxiliary_loss": 0.004265990573912859, "train/entity_pixel_l1": 0.024959683418273926, "train/entity_pixel_edge": 0.012485100887715816, "train/player_pixel_l1": 0.03831017017364502, "train/player_pixel_edge": 0.017398955300450325, "train/health_pixel_l1": 0.03899902105331421, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724038124084473, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7230, "train/total_loss": 0.7527844905853271, "train/flow_loss": 0.21172188222408295, "train/player_flow_loss": 0.536626398563385, "train/auxiliary_loss": 0.004436212126165628, "train/entity_pixel_l1": 0.025146160274744034, "train/entity_pixel_edge": 0.01238101627677679, "train/player_pixel_l1": 0.039827387779951096, "train/player_pixel_edge": 0.018138941377401352, "train/health_pixel_l1": 0.036802731454372406, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723061561584473, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7240, "train/total_loss": 0.6583924889564514, "train/flow_loss": 0.2206266224384308, "train/player_flow_loss": 0.43382319808006287, "train/auxiliary_loss": 0.003942674025893211, "train/entity_pixel_l1": 0.02382390759885311, "train/entity_pixel_edge": 0.011716463603079319, "train/player_pixel_l1": 0.0361725315451622, "train/player_pixel_edge": 0.013016821816563606, "train/health_pixel_l1": 0.04078817367553711, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723112106323242, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7250, "train/total_loss": 0.7497276067733765, "train/flow_loss": 0.2679004669189453, "train/player_flow_loss": 0.4780421257019043, "train/auxiliary_loss": 0.003785058856010437, "train/entity_pixel_l1": 0.028108175843954086, "train/entity_pixel_edge": 0.013723183423280716, "train/player_pixel_l1": 0.03411676734685898, "train/player_pixel_edge": 0.014935292303562164, "train/health_pixel_l1": 0.04006759822368622, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724363327026367, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7260, "train/total_loss": 0.6318573951721191, "train/flow_loss": 0.17157244682312012, "train/player_flow_loss": 0.45679134130477905, "train/auxiliary_loss": 0.0034936065785586834, "train/entity_pixel_l1": 0.022342950105667114, "train/entity_pixel_edge": 0.012528665363788605, "train/player_pixel_l1": 0.030693938955664635, "train/player_pixel_edge": 0.016968507319688797, "train/health_pixel_l1": 0.03925804793834686, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724363327026367, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7270, "train/total_loss": 0.7256112694740295, "train/flow_loss": 0.2745913565158844, "train/player_flow_loss": 0.4473344385623932, "train/auxiliary_loss": 0.003685449482873082, "train/entity_pixel_l1": 0.023128673434257507, "train/entity_pixel_edge": 0.013169955462217331, "train/player_pixel_l1": 0.032636113464832306, "train/player_pixel_edge": 0.01687353290617466, "train/health_pixel_l1": 0.03936222195625305, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724637985229492, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7280, "train/total_loss": 0.6625636219978333, "train/flow_loss": 0.19355405867099762, "train/player_flow_loss": 0.4649573266506195, "train/auxiliary_loss": 0.004052283242344856, "train/entity_pixel_l1": 0.023038052022457123, "train/entity_pixel_edge": 0.011983639560639858, "train/player_pixel_l1": 0.036312468349933624, "train/player_pixel_edge": 0.01684143766760826, "train/health_pixel_l1": 0.04037674143910408, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.717844009399414, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7290, "train/total_loss": 0.6314188241958618, "train/flow_loss": 0.19341956079006195, "train/player_flow_loss": 0.43373358249664307, "train/auxiliary_loss": 0.004265677183866501, "train/entity_pixel_l1": 0.02190866880118847, "train/entity_pixel_edge": 0.012257897295057774, "train/player_pixel_l1": 0.03746047616004944, "train/player_pixel_edge": 0.020785193890333176, "train/health_pixel_l1": 0.04274194687604904, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724164962768555, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7300, "train/total_loss": 0.5985969305038452, "train/flow_loss": 0.20044900476932526, "train/player_flow_loss": 0.3946648836135864, "train/auxiliary_loss": 0.0034830132499337196, "train/entity_pixel_l1": 0.020293740555644035, "train/entity_pixel_edge": 0.01197564136236906, "train/player_pixel_l1": 0.030397240072488785, "train/player_pixel_edge": 0.017731565982103348, "train/health_pixel_l1": 0.038635775446891785, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.724451065063477, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7310, "train/total_loss": 0.47711652517318726, "train/flow_loss": 0.14670182764530182, "train/player_flow_loss": 0.3273913264274597, "train/auxiliary_loss": 0.0030233508441597223, "train/entity_pixel_l1": 0.022780397906899452, "train/entity_pixel_edge": 0.011339492164552212, "train/player_pixel_l1": 0.026020711287856102, "train/player_pixel_edge": 0.016851183027029037, "train/health_pixel_l1": 0.037351276725530624, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723112106323242, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7320, "train/total_loss": 0.6435187458992004, "train/flow_loss": 0.19787539541721344, "train/player_flow_loss": 0.4416421055793762, "train/auxiliary_loss": 0.004001299850642681, "train/entity_pixel_l1": 0.023693205788731575, "train/entity_pixel_edge": 0.012649105861783028, "train/player_pixel_l1": 0.03552311658859253, "train/player_pixel_edge": 0.017959516495466232, "train/health_pixel_l1": 0.03895771875977516, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72355842590332, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7330, "train/total_loss": 0.7382825613021851, "train/flow_loss": 0.21899732947349548, "train/player_flow_loss": 0.5143383741378784, "train/auxiliary_loss": 0.0049468399956822395, "train/entity_pixel_l1": 0.02031528577208519, "train/entity_pixel_edge": 0.0110853286460042, "train/player_pixel_l1": 0.044863440096378326, "train/player_pixel_edge": 0.018419833853840828, "train/health_pixel_l1": 0.03725283965468407, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72340202331543, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7340, "train/total_loss": 0.5172487497329712, "train/flow_loss": 0.16176533699035645, "train/player_flow_loss": 0.3513302206993103, "train/auxiliary_loss": 0.0041531752794981, "train/entity_pixel_l1": 0.023516131564974785, "train/entity_pixel_edge": 0.012042272835969925, "train/player_pixel_l1": 0.03727205470204353, "train/player_pixel_edge": 0.017038773745298386, "train/health_pixel_l1": 0.036889009177684784, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.718517303466797, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7350, "train/total_loss": 0.6037335395812988, "train/flow_loss": 0.18503417074680328, "train/player_flow_loss": 0.41504019498825073, "train/auxiliary_loss": 0.0036592159885913134, "train/entity_pixel_l1": 0.020623505115509033, "train/entity_pixel_edge": 0.012550659477710724, "train/player_pixel_l1": 0.031815700232982635, "train/player_pixel_edge": 0.01910584047436714, "train/health_pixel_l1": 0.04140926152467728, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.723466873168945, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}
+{"step": 7360, "train/total_loss": 0.7072920799255371, "train/flow_loss": 0.22356954216957092, "train/player_flow_loss": 0.47797802090644836, "train/auxiliary_loss": 0.005744491703808308, "train/entity_pixel_l1": 0.03501550853252411, "train/entity_pixel_edge": 0.011468702927231789, "train/player_pixel_l1": 0.05372101068496704, "train/player_pixel_edge": 0.014895620755851269, "train/health_pixel_l1": 0.044232286512851715, "train/player_identity_loss": 0.0, "train/player_identity_similarity": 0.0, "train/player_identity_ranking_accuracy": 0.0, "train/counterfactual_player_difference_loss": 0.0, "train/learning_rate": 1e-05, "train/lr_backbone_and_conditions": 1e-05, "train/lr_appearance": 5e-05, "train/lr_joint_patch": 1e-05, "train/cuda_memory_allocated_max_gib": 9.72395133972168, "train/cuda_peak_allocated_max_gib": 20.021989822387695, "train/cuda_peak_reserved_max_gib": 21.400390625, "train/counterfactual_step": 0.0}

m3/m3_simple_fulltrain_player_flow_step7365/validation.jsonl

@@ -0,0 +1,3 @@
+{"step": 6750, "total_loss": 0.8140180110931396, "flow_loss": 0.32706335186958313, "player_flow_loss": 0.48318836092948914, "auxiliary_loss": 0.003766144858673215, "entity_pixel_l1": 0.03157222643494606, "entity_pixel_edge": 0.014456245116889477, "player_pixel_l1": 0.03361259400844574, "player_pixel_edge": 0.016195405274629593, "health_pixel_l1": 0.047234613448381424, "player_identity_loss": 0.0, "player_identity_similarity": 0.0, "player_identity_ranking_accuracy": 0.0, "counterfactual_player_difference_loss": 0.0}
+{"step": 7000, "total_loss": 0.7994883060455322, "flow_loss": 0.32039517164230347, "player_flow_loss": 0.47531163692474365, "auxiliary_loss": 0.003781450679525733, "entity_pixel_l1": 0.03138159215450287, "entity_pixel_edge": 0.01452136691659689, "player_pixel_l1": 0.033745598047971725, "player_pixel_edge": 0.01627562940120697, "health_pixel_l1": 0.04682464152574539, "player_identity_loss": 0.0, "player_identity_similarity": 0.0, "player_identity_ranking_accuracy": 0.0, "counterfactual_player_difference_loss": 0.0}
+{"step": 7250, "total_loss": 0.7877868413925171, "flow_loss": 0.3170263171195984, "player_flow_loss": 0.4670434892177582, "auxiliary_loss": 0.003716996405273676, "entity_pixel_l1": 0.030641857534646988, "entity_pixel_edge": 0.014442835003137589, "player_pixel_l1": 0.03313995152711868, "player_pixel_edge": 0.016120050102472305, "health_pixel_l1": 0.04664304852485657, "player_identity_loss": 0.0, "player_identity_similarity": 0.0, "player_identity_ranking_accuracy": 0.0, "counterfactual_player_difference_loss": 0.0}

m3/m3_simple_fulltrain_player_flow_step7365/wandb_run.json

@@ -0,0 +1,6 @@
+{
+  "id": "fhh63zsg",
+  "url": "https://wandb.ai/ckx23-tsinghua-university/plot-m3/runs/fhh63zsg",
+  "entity": "ckx23-tsinghua-university",
+  "project": "plot-m3"
+}