auto-sync 2026-07-02T13:48:50Z scratch/experiments/dovla_h16_policy_ckpt_runs (part 6)

.gitattributes

@@ -290,3 +290,36 @@ scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_1/trans
 scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_1/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1.json filter=lfs diff=lfs merge=lfs -text
 scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_1/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1.json filter=lfs diff=lfs merge=lfs -text
 scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_1/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_0_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_1_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_2_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_3_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_4_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_5_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_6_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_7_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_0_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_1_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_2_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_3_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_4_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_5_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_6_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_7_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_0_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_1_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_2_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_3_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_4_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_5_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_6_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_7_of_8.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/transport_field_targets_k4compose_dropnmnoop_grid035040045_margin0p20_noopbonus0p03_nmchallenger0p01_b8_sharded8_v2.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/retrieval_residual_teacher_targets_k4compose_nmchallenger/seed_0/train_targets.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/retrieval_residual_teacher_targets_k4compose_nmchallenger/seed_1/train_targets.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/retrieval_residual_teacher_targets_k4compose_nmchallenger/seed_2/train_targets.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/targets/retrieval_residual_scale0p35_safe_types_teacher_seed2_allmap.json filter=lfs diff=lfs merge=lfs -text
+scratch/experiments/dovla_h16_policy_ckpt_runs/targets/retrieval_residual_scale0p35_safe_types_teacher_seed2_allmap_v2.json filter=lfs diff=lfs merge=lfs -text

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_0_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_1_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_2_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_3_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_4_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_5_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_6_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1_shard_7_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_0_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_1_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_2_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_3_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_4_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_5_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_6_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1_shard_7_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_0_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_1_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_2_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_3_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_4_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_5_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_6_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/shards/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2_shard_7_of_8.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/transport_field_targets_k4compose_dropnmnoop_grid035040045_margin0p20_noopbonus0p03_nmchallenger0p01_b8_sharded8_v2.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b12_sharded8_v1.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/transport_field_targets_k6compose_dropnmnoop_dropnoopwg_grid035040045_margin0p00_clean_b24_sharded8_v1.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5/seed_2/transport_field_targets_k6compose_dropnmnoop_grid035040045_margin0p00_clean_b12_sharded8_v2.json

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_0/best.pt

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_0/best_policy.pt

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_0/latest.pt

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_0/metrics.json

@@ -0,0 +1,1582 @@
+{
+  "best": {
+    "bc_loss": 0.1395697204603089,
+    "field_effect_loss": 0.009249651183684668,
+    "field_potential_loss": 0.2649960385428535,
+    "field_preference_loss": 0.5072847704092661,
+    "lattice_edges": 3833.3333333333335,
+    "progress_mae": 0.1867915623717838,
+    "proposal_loss": 3.03697289736091e-06,
+    "rank_acc": 0.8329584466086494,
+    "rank_loss": 0.5072785168886185,
+    "regret_mae": 0.39045696788363987,
+    "success_accuracy": 0.8741144405470954,
+    "total_loss": 1.55989404519399
+  },
+  "best_policy": {
+    "bc_loss": 0.1388491830892033,
+    "field_effect_loss": 0.009226797407286035,
+    "field_potential_loss": 0.2551382887694571,
+    "field_preference_loss": 0.516539043850369,
+    "lattice_edges": 3833.3333333333335,
+    "progress_mae": 0.1904102149936888,
+    "proposal_loss": 3.2502353709181384e-06,
+    "rank_acc": 0.8304381403658125,
+    "rank_loss": 0.516533311870363,
+    "regret_mae": 0.3661145286427604,
+    "success_accuracy": 0.8777371744314829,
+    "total_loss": 1.557318925857544
+  },
+  "history": [
+    {
+      "epoch": 1,
+      "train": {
+        "bc_loss": 0.32647183164954185,
+        "field_effect_loss": 0.01198829308203939,
+        "field_potential_loss": 0.3826601058244705,
+        "field_preference_loss": 0.5572003531787131,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.20559305883944035,
+        "proposal_loss": 0.0006480094748945299,
+        "rank_acc": 0.7668617889285088,
+        "rank_loss": 0.5571942379077276,
+        "regret_mae": 0.6372969270580344,
+        "success_accuracy": 0.8552538231015205,
+        "total_loss": 2.6922242641448975
+      },
+      "val": {
+        "bc_loss": 0.24306229170825747,
+        "field_effect_loss": 0.01052365560705463,
+        "field_potential_loss": 0.34387966990470886,
+        "field_preference_loss": 0.5559452407889895,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20100927352905273,
+        "proposal_loss": 0.00010651227057678625,
+        "rank_acc": 0.7839428583780924,
+        "rank_loss": 0.5559457606739469,
+        "regret_mae": 0.6408805151780447,
+        "success_accuracy": 0.8590949806902144,
+        "total_loss": 2.2280810144212513
+      }
+    },
+    {
+      "epoch": 2,
+      "train": {
+        "bc_loss": 0.20886539460884201,
+        "field_effect_loss": 0.010721629847668938,
+        "field_potential_loss": 0.32831621087259716,
+        "field_preference_loss": 0.5356637558175458,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.192453161916799,
+        "proposal_loss": 7.547787592759253e-05,
+        "rank_acc": 0.7973566568560071,
+        "rank_loss": 0.5356602540446652,
+        "regret_mae": 0.524406803978814,
+        "success_accuracy": 0.8687212044994036,
+        "total_loss": 2.0130514601866403
+      },
+      "val": {
+        "bc_loss": 0.2088448657757706,
+        "field_effect_loss": 0.010271870686362186,
+        "field_potential_loss": 0.30628737310568493,
+        "field_preference_loss": 0.5442188150352902,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20224370641840828,
+        "proposal_loss": 4.840822394827329e-05,
+        "rank_acc": 0.8023176160123613,
+        "rank_loss": 0.5442166129748026,
+        "regret_mae": 0.5458454257912106,
+        "success_accuracy": 0.8565923207336001,
+        "total_loss": 2.0040130151642694
+      }
+    },
+    {
+      "epoch": 3,
+      "train": {
+        "bc_loss": 0.1888708436033792,
+        "field_effect_loss": 0.01053104166769319,
+        "field_potential_loss": 0.3023387195749415,
+        "field_preference_loss": 0.5275019485917356,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1921633780002594,
+        "proposal_loss": 4.097332382318402e-05,
+        "rank_acc": 0.8051786116427846,
+        "rank_loss": 0.5274932719767094,
+        "regret_mae": 0.4670240750743283,
+        "success_accuracy": 0.8687858912679884,
+        "total_loss": 1.8751557899845972
+      },
+      "val": {
+        "bc_loss": 0.1905435079501735,
+        "field_effect_loss": 0.010035918715099493,
+        "field_potential_loss": 0.3004525717761781,
+        "field_preference_loss": 0.5400237672858768,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19465689029958513,
+        "proposal_loss": 3.0318424958548145e-05,
+        "rank_acc": 0.8025992777612474,
+        "rank_loss": 0.5400202870368958,
+        "regret_mae": 0.4824800259537167,
+        "success_accuracy": 0.8568093346224891,
+        "total_loss": 1.900115708510081
+      }
+    },
+    {
+      "epoch": 4,
+      "train": {
+        "bc_loss": 0.17609142139554024,
+        "field_effect_loss": 0.010249738581478596,
+        "field_potential_loss": 0.28818330023851657,
+        "field_preference_loss": 0.5228900822500387,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.19246476329863071,
+        "proposal_loss": 2.5494585088381427e-05,
+        "rank_acc": 0.8117518507772021,
+        "rank_loss": 0.5228802652822601,
+        "regret_mae": 0.44389980990025735,
+        "success_accuracy": 0.869324252837234,
+        "total_loss": 1.79043228758706
+      },
+      "val": {
+        "bc_loss": 0.18543473589751455,
+        "field_effect_loss": 0.009758494432187743,
+        "field_potential_loss": 0.2895740204387241,
+        "field_preference_loss": 0.5251913037565019,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18942881872256598,
+        "proposal_loss": 2.2599307360198712e-05,
+        "rank_acc": 0.8075593047671847,
+        "rank_loss": 0.5251863367027707,
+        "regret_mae": 0.4111841453446282,
+        "success_accuracy": 0.8696516553560892,
+        "total_loss": 1.843348052766588
+      }
+    },
+    {
+      "epoch": 5,
+      "train": {
+        "bc_loss": 0.16067334078252316,
+        "field_effect_loss": 0.010109223179622658,
+        "field_potential_loss": 0.2809416730370786,
+        "field_preference_loss": 0.5203770171436999,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18999037043088013,
+        "proposal_loss": 1.7667226112886663e-05,
+        "rank_acc": 0.8150273602869775,
+        "rank_loss": 0.5203685433500342,
+        "regret_mae": 0.4251602436933253,
+        "success_accuracy": 0.8731553827722868,
+        "total_loss": 1.702083130677541
+      },
+      "val": {
+        "bc_loss": 0.17296147346496582,
+        "field_effect_loss": 0.009765082452860143,
+        "field_potential_loss": 0.3102547542916404,
+        "field_preference_loss": 0.5111219750510322,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1860836504234208,
+        "proposal_loss": 1.5547995038408164e-05,
+        "rank_acc": 0.8132495151625739,
+        "rank_loss": 0.5111174053615994,
+        "regret_mae": 0.45361913409498,
+        "success_accuracy": 0.8746464749177297,
+        "total_loss": 1.7821607457266913
+      }
+    },
+    {
+      "epoch": 6,
+      "train": {
+        "bc_loss": 0.15348759107291698,
+        "field_effect_loss": 0.010033887164253328,
+        "field_potential_loss": 0.27619009568459457,
+        "field_preference_loss": 0.5180389636920558,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.19086008125709164,
+        "proposal_loss": 1.3807272188791507e-05,
+        "rank_acc": 0.8183441774712669,
+        "rank_loss": 0.518029462132189,
+        "regret_mae": 0.42153678875830436,
+        "success_accuracy": 0.8763730302453041,
+        "total_loss": 1.657825180225902
+      },
+      "val": {
+        "bc_loss": 0.17016205398572815,
+        "field_effect_loss": 0.009703818191256788,
+        "field_potential_loss": 0.28311825378073585,
+        "field_preference_loss": 0.5273637076218923,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19934751010603374,
+        "proposal_loss": 1.2484028957260307e-05,
+        "rank_acc": 0.8130956457720863,
+        "rank_loss": 0.5273616645071242,
+        "regret_mae": 0.44492850369877285,
+        "success_accuracy": 0.8610586060418023,
+        "total_loss": 1.7618852721320257
+      }
+    },
+    {
+      "epoch": 7,
+      "train": {
+        "bc_loss": 0.14851212729182509,
+        "field_effect_loss": 0.009917672904622223,
+        "field_potential_loss": 0.27055538776848054,
+        "field_preference_loss": 0.5160685694879956,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18956360303693348,
+        "proposal_loss": 1.0902919105622763e-05,
+        "rank_acc": 0.8231632626718945,
+        "rank_loss": 0.5160607318911288,
+        "regret_mae": 0.40903862772716415,
+        "success_accuracy": 0.8718783391846551,
+        "total_loss": 1.6247840192582872
+      },
+      "val": {
+        "bc_loss": 0.1598210045033031,
+        "field_effect_loss": 0.00955163718511661,
+        "field_potential_loss": 0.29387929538885754,
+        "field_preference_loss": 0.505281569229232,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18717868626117706,
+        "proposal_loss": 9.630582301017259e-06,
+        "rank_acc": 0.8169737358887991,
+        "rank_loss": 0.5052787297301822,
+        "regret_mae": 0.40659383104907143,
+        "success_accuracy": 0.8718182941277822,
+        "total_loss": 1.6939869854185317
+      }
+    },
+    {
+      "epoch": 8,
+      "train": {
+        "bc_loss": 0.13903531928857168,
+        "field_effect_loss": 0.009911321789129741,
+        "field_potential_loss": 0.26990564664204914,
+        "field_preference_loss": 0.5158376747535335,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.19043918574849764,
+        "proposal_loss": 8.923169262844669e-06,
+        "rank_acc": 0.8229603742559751,
+        "rank_loss": 0.5158255062997341,
+        "regret_mae": 0.41572194173932076,
+        "success_accuracy": 0.8710144567820761,
+        "total_loss": 1.576402520140012
+      },
+      "val": {
+        "bc_loss": 0.16312216139501995,
+        "field_effect_loss": 0.009572485979232524,
+        "field_potential_loss": 0.2609928556614452,
+        "field_preference_loss": 0.5401404466893938,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19842411246564653,
+        "proposal_loss": 7.900755791625772e-06,
+        "rank_acc": 0.815243317021264,
+        "rank_loss": 0.5401360988616943,
+        "regret_mae": 0.435974571439955,
+        "success_accuracy": 0.8621471756034427,
+        "total_loss": 1.7176929447386
+      }
+    },
+    {
+      "epoch": 9,
+      "train": {
+        "bc_loss": 0.13357605247033966,
+        "field_effect_loss": 0.009837586956564337,
+        "field_potential_loss": 0.2669225597961081,
+        "field_preference_loss": 0.513505751060115,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18997566836575666,
+        "proposal_loss": 7.133084510415857e-06,
+        "rank_acc": 0.8251901939511299,
+        "rank_loss": 0.5134942394991716,
+        "regret_mae": 0.3984128021531635,
+        "success_accuracy": 0.8729508883423276,
+        "total_loss": 1.5429423385196261
+      },
+      "val": {
+        "bc_loss": 0.1578697549800078,
+        "field_effect_loss": 0.00955356854117579,
+        "field_potential_loss": 0.2772790938615799,
+        "field_preference_loss": 0.5219141791264216,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18713324020306268,
+        "proposal_loss": 6.961688136976187e-06,
+        "rank_acc": 0.8149134582943387,
+        "rank_loss": 0.5219109637869729,
+        "regret_mae": 0.4014560696151521,
+        "success_accuracy": 0.8725778427388933,
+        "total_loss": 1.6879634724722967
+      }
+    },
+    {
+      "epoch": 10,
+      "train": {
+        "bc_loss": 0.13299925925417078,
+        "field_effect_loss": 0.009794742755022727,
+        "field_potential_loss": 0.25979059458606774,
+        "field_preference_loss": 0.5145298677186171,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18957360544138485,
+        "proposal_loss": 6.454507462826263e-06,
+        "rank_acc": 0.8264916928278075,
+        "rank_loss": 0.5145187692509757,
+        "regret_mae": 0.4029802485472626,
+        "success_accuracy": 0.8736165372861756,
+        "total_loss": 1.5333838396602206
+      },
+      "val": {
+        "bc_loss": 0.16119847405287954,
+        "field_effect_loss": 0.00951479985896084,
+        "field_potential_loss": 0.2622404272357623,
+        "field_preference_loss": 0.5294790632194943,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19883924888239968,
+        "proposal_loss": 6.655631194816023e-06,
+        "rank_acc": 0.8227982653511895,
+        "rank_loss": 0.5294736921787262,
+        "regret_mae": 0.43525839679771,
+        "success_accuracy": 0.8656158977084689,
+        "total_loss": 1.6955701775021024
+      }
+    },
+    {
+      "epoch": 11,
+      "train": {
+        "bc_loss": 0.12502337174697054,
+        "field_effect_loss": 0.00971597424035685,
+        "field_potential_loss": 0.2546710566514068,
+        "field_preference_loss": 0.5113476692802377,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.19071267400350836,
+        "proposal_loss": 5.670387932039173e-06,
+        "rank_acc": 0.8312145951721404,
+        "rank_loss": 0.5113350339233875,
+        "regret_mae": 0.3956419920755757,
+        "success_accuracy": 0.8743489591611756,
+        "total_loss": 1.484339983926879
+      },
+      "val": {
+        "bc_loss": 0.1603577472269535,
+        "field_effect_loss": 0.009401841192609735,
+        "field_potential_loss": 0.2561184482442008,
+        "field_preference_loss": 0.5291704701052772,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19469515648153093,
+        "proposal_loss": 5.696274102875678e-06,
+        "rank_acc": 0.82403873735004,
+        "rank_loss": 0.5291660924752554,
+        "regret_mae": 0.3835206561618381,
+        "success_accuracy": 0.8621436754862467,
+        "total_loss": 1.6842918793360393
+      }
+    },
+    {
+      "epoch": 12,
+      "train": {
+        "bc_loss": 0.12039409075967139,
+        "field_effect_loss": 0.009685435998512225,
+        "field_potential_loss": 0.25417477182216114,
+        "field_preference_loss": 0.5097686126828194,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1894871350377798,
+        "proposal_loss": 4.863858072844272e-06,
+        "rank_acc": 0.8323832435740365,
+        "rank_loss": 0.5097574852406979,
+        "regret_mae": 0.3824283426834477,
+        "success_accuracy": 0.8750939004951053,
+        "total_loss": 1.4589199506574206
+      },
+      "val": {
+        "bc_loss": 0.1494932501680321,
+        "field_effect_loss": 0.009295274839840002,
+        "field_potential_loss": 0.27344995985428494,
+        "field_preference_loss": 0.5074194338586595,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18854867584175533,
+        "proposal_loss": 4.777456145651134e-06,
+        "rank_acc": 0.8218355245060391,
+        "rank_loss": 0.5074140412939919,
+        "regret_mae": 0.4008950806326336,
+        "success_accuracy": 0.8757280442449782,
+        "total_loss": 1.6209543347358704
+      }
+    },
+    {
+      "epoch": 13,
+      "train": {
+        "bc_loss": 0.11936052133225733,
+        "field_effect_loss": 0.009638249777102223,
+        "field_potential_loss": 0.2496584802865982,
+        "field_preference_loss": 0.5088633861806657,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18991429482897124,
+        "proposal_loss": 4.54581102202509e-06,
+        "rank_acc": 0.8346696653299861,
+        "rank_loss": 0.5088501630557908,
+        "regret_mae": 0.3825601227581501,
+        "success_accuracy": 0.8766860316197077,
+        "total_loss": 1.4474417550696268
+      },
+      "val": {
+        "bc_loss": 0.15197822120454577,
+        "field_effect_loss": 0.009337011336659392,
+        "field_potential_loss": 0.2695453382200665,
+        "field_preference_loss": 0.5135077403651344,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1889910822113355,
+        "proposal_loss": 4.753038385590318e-06,
+        "rank_acc": 0.8221405380302005,
+        "rank_loss": 0.5135033329327902,
+        "regret_mae": 0.37687303788132137,
+        "success_accuracy": 0.8728998634550307,
+        "total_loss": 1.636679232120514
+      }
+    },
+    {
+      "epoch": 14,
+      "train": {
+        "bc_loss": 0.1129105842879249,
+        "field_effect_loss": 0.009580075592061298,
+        "field_potential_loss": 0.24452892297671902,
+        "field_preference_loss": 0.5078012239601877,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1887665055692196,
+        "proposal_loss": 4.032907794579741e-06,
+        "rank_acc": 0.838013619184494,
+        "rank_loss": 0.5077874006496536,
+        "regret_mae": 0.3774424199428823,
+        "success_accuracy": 0.8768967853652107,
+        "total_loss": 1.4083480768733554
+      },
+      "val": {
+        "bc_loss": 0.14547683050235113,
+        "field_effect_loss": 0.009272080897870991,
+        "field_potential_loss": 0.2586834281682968,
+        "field_preference_loss": 0.5161848929193285,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1905267900890774,
+        "proposal_loss": 3.877517428918509e-06,
+        "rank_acc": 0.8254512184196048,
+        "rank_loss": 0.5161795566479365,
+        "regret_mae": 0.3768312517139647,
+        "success_accuracy": 0.87029919690556,
+        "total_loss": 1.5969177219602797
+      }
+    },
+    {
+      "epoch": 15,
+      "train": {
+        "bc_loss": 0.1082119130425983,
+        "field_effect_loss": 0.009549448822831942,
+        "field_potential_loss": 0.24023904485834968,
+        "field_preference_loss": 0.5052970697482427,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18983113124138778,
+        "proposal_loss": 3.7523693063374394e-06,
+        "rank_acc": 0.838187698688772,
+        "rank_loss": 0.5052847063375844,
+        "regret_mae": 0.3717968459758494,
+        "success_accuracy": 0.8795969386895498,
+        "total_loss": 1.3774204270707235
+      },
+      "val": {
+        "bc_loss": 0.1475692842569616,
+        "field_effect_loss": 0.009338218537676666,
+        "field_potential_loss": 0.27476220577955246,
+        "field_preference_loss": 0.5110197265942892,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1934267116917504,
+        "proposal_loss": 3.929910942840959e-06,
+        "rank_acc": 0.8201194438669417,
+        "rank_loss": 0.5110183738999896,
+        "regret_mae": 0.40272161695692277,
+        "success_accuracy": 0.8756335377693176,
+        "total_loss": 1.6152734955151875
+      }
+    },
+    {
+      "epoch": 16,
+      "train": {
+        "bc_loss": 0.10503616194344229,
+        "field_effect_loss": 0.009528841894482158,
+        "field_potential_loss": 0.237404505825705,
+        "field_preference_loss": 0.5044695834318796,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1907089567846722,
+        "proposal_loss": 3.8791208807096054e-06,
+        "rank_acc": 0.8395452284150653,
+        "rank_loss": 0.5044577510820495,
+        "regret_mae": 0.37322405229012173,
+        "success_accuracy": 0.8792067310876317,
+        "total_loss": 1.3574828174379137
+      },
+      "val": {
+        "bc_loss": 0.1550265136692259,
+        "field_effect_loss": 0.009431856761996945,
+        "field_potential_loss": 0.27428053071101505,
+        "field_preference_loss": 0.5154997259378433,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18637417339616352,
+        "proposal_loss": 3.936731104658975e-06,
+        "rank_acc": 0.8215479453404745,
+        "rank_loss": 0.5154954476488961,
+        "regret_mae": 0.37237905462582904,
+        "success_accuracy": 0.8731273776955075,
+        "total_loss": 1.657362871699863
+      }
+    },
+    {
+      "epoch": 17,
+      "train": {
+        "bc_loss": 0.10400018199450439,
+        "field_effect_loss": 0.009505713067483157,
+        "field_potential_loss": 0.2361969355907705,
+        "field_preference_loss": 0.504442010488775,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18794113832215467,
+        "proposal_loss": 3.6243958700449893e-06,
+        "rank_acc": 0.8412649432818095,
+        "rank_loss": 0.5044302298790879,
+        "regret_mae": 0.3686659493380123,
+        "success_accuracy": 0.8776417275269827,
+        "total_loss": 1.351132548517651
+      },
+      "val": {
+        "bc_loss": 0.14389586986766922,
+        "field_effect_loss": 0.009288348438632157,
+        "field_potential_loss": 0.26957807689905167,
+        "field_preference_loss": 0.5120650182167689,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18726851128869587,
+        "proposal_loss": 3.6443942942949234e-06,
+        "rank_acc": 0.8220728039741516,
+        "rank_loss": 0.5120610313283073,
+        "regret_mae": 0.3811516496870253,
+        "success_accuracy": 0.8767046067449782,
+        "total_loss": 1.5936155849032931
+      }
+    },
+    {
+      "epoch": 18,
+      "train": {
+        "bc_loss": 0.09931996138766408,
+        "field_effect_loss": 0.009506737173069268,
+        "field_potential_loss": 0.23368702083826065,
+        "field_preference_loss": 0.5029443233377404,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1887050610449579,
+        "proposal_loss": 3.708235220124657e-06,
+        "rank_acc": 0.8427573326561186,
+        "rank_loss": 0.5029311693376965,
+        "regret_mae": 0.36805426329374313,
+        "success_accuracy": 0.8789771969119707,
+        "total_loss": 1.323204497496287
+      },
+      "val": {
+        "bc_loss": 0.13926093321707514,
+        "field_effect_loss": 0.0093413472123858,
+        "field_potential_loss": 0.27169622977574664,
+        "field_preference_loss": 0.515564438369539,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18942548086245856,
+        "proposal_loss": 3.243226640531955e-06,
+        "rank_acc": 0.8246809211042192,
+        "rank_loss": 0.5155611087878545,
+        "regret_mae": 0.41602158215310836,
+        "success_accuracy": 0.8767046067449782,
+        "total_loss": 1.5757743186420865
+      }
+    },
+    {
+      "epoch": 19,
+      "train": {
+        "bc_loss": 0.09842619956988427,
+        "field_effect_loss": 0.009522452969880154,
+        "field_potential_loss": 0.23635326657030317,
+        "field_preference_loss": 0.5040057773391405,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18746552885406548,
+        "proposal_loss": 3.3772429080474667e-06,
+        "rank_acc": 0.8419533396760622,
+        "rank_loss": 0.5039918402002918,
+        "regret_mae": 0.3702448830008507,
+        "success_accuracy": 0.8790105837914679,
+        "total_loss": 1.3225920697053273
+      },
+      "val": {
+        "bc_loss": 0.1457436088886526,
+        "field_effect_loss": 0.009281905988852182,
+        "field_potential_loss": 0.2571987269653214,
+        "field_preference_loss": 0.5196319156222873,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19150949269533157,
+        "proposal_loss": 3.5643024855542335e-06,
+        "rank_acc": 0.8250452015135024,
+        "rank_loss": 0.5196255279911889,
+        "regret_mae": 0.3816456198692322,
+        "success_accuracy": 0.8722628222571479,
+        "total_loss": 1.5984870394070942
+      }
+    },
+    {
+      "epoch": 20,
+      "train": {
+        "bc_loss": 0.09496854147356418,
+        "field_effect_loss": 0.009445087466802862,
+        "field_potential_loss": 0.2306023163514005,
+        "field_preference_loss": 0.5018902810083495,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18800238851043913,
+        "proposal_loss": 3.3296213959804136e-06,
+        "rank_acc": 0.8441196895307965,
+        "rank_loss": 0.5018755462434557,
+        "regret_mae": 0.3662330222626527,
+        "success_accuracy": 0.8797450926568773,
+        "total_loss": 1.2966503318813112
+      },
+      "val": {
+        "bc_loss": 0.14245796493358082,
+        "field_effect_loss": 0.0092617969494313,
+        "field_potential_loss": 0.26488720460070503,
+        "field_preference_loss": 0.5171391897731357,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1923482103480233,
+        "proposal_loss": 3.2551362968054795e-06,
+        "rank_acc": 0.8277582426865896,
+        "rank_loss": 0.5171360853645537,
+        "regret_mae": 0.3922466884056727,
+        "success_accuracy": 0.8725813428560892,
+        "total_loss": 1.587904281086392
+      }
+    },
+    {
+      "epoch": 21,
+      "train": {
+        "bc_loss": 0.09108199919056562,
+        "field_effect_loss": 0.009438617310176292,
+        "field_potential_loss": 0.2300427866478761,
+        "field_preference_loss": 0.5011279740267329,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18758674007323053,
+        "proposal_loss": 3.422313388531014e-06,
+        "rank_acc": 0.8458103885253271,
+        "rank_loss": 0.5011149193677638,
+        "regret_mae": 0.35920447959668106,
+        "success_accuracy": 0.8806089750594563,
+        "total_loss": 1.2752653857072194
+      },
+      "val": {
+        "bc_loss": 0.14189000634683502,
+        "field_effect_loss": 0.009239565710433655,
+        "field_potential_loss": 0.2593420238958465,
+        "field_preference_loss": 0.5119498438305325,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18742902328570685,
+        "proposal_loss": 3.981592638208693e-06,
+        "rank_acc": 0.8306096957789527,
+        "rank_loss": 0.5119451383749644,
+        "regret_mae": 0.39159053729640114,
+        "success_accuracy": 0.873792419830958,
+        "total_loss": 1.5721659594111972
+      }
+    },
+    {
+      "epoch": 22,
+      "train": {
+        "bc_loss": 0.08891944157787496,
+        "field_effect_loss": 0.00942038031320812,
+        "field_potential_loss": 0.22684395665095913,
+        "field_preference_loss": 0.4997193246252007,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18747266092234188,
+        "proposal_loss": 3.4626974922957743e-06,
+        "rank_acc": 0.8477694797847006,
+        "rank_loss": 0.49970687304933864,
+        "regret_mae": 0.3659054934978485,
+        "success_accuracy": 0.8802125073141522,
+        "total_loss": 1.2595618516206741
+      },
+      "val": {
+        "bc_loss": 0.14417342758840984,
+        "field_effect_loss": 0.009213971465619074,
+        "field_potential_loss": 0.27098352048132157,
+        "field_preference_loss": 0.5044561558299594,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18478062583340538,
+        "proposal_loss": 3.576060572009333e-06,
+        "rank_acc": 0.829954551325904,
+        "rank_loss": 0.5044514404402839,
+        "regret_mae": 0.3868408020999696,
+        "success_accuracy": 0.8741319444444444,
+        "total_loss": 1.5869588785701327
+      }
+    },
+    {
+      "epoch": 23,
+      "train": {
+        "bc_loss": 0.08818179834634066,
+        "field_effect_loss": 0.009438060683250014,
+        "field_potential_loss": 0.22758339966336885,
+        "field_preference_loss": 0.49927598817480934,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18710674718022346,
+        "proposal_loss": 3.2211552410091017e-06,
+        "rank_acc": 0.847735991080602,
+        "rank_loss": 0.49926325554649037,
+        "regret_mae": 0.36335276977883446,
+        "success_accuracy": 0.8816335308882925,
+        "total_loss": 1.2563018649816513
+      },
+      "val": {
+        "bc_loss": 0.140900277843078,
+        "field_effect_loss": 0.009297234756458137,
+        "field_potential_loss": 0.26961542748742634,
+        "field_preference_loss": 0.5097482767370012,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18338180581728616,
+        "proposal_loss": 2.9214355284542157e-06,
+        "rank_acc": 0.8293458819389343,
+        "rank_loss": 0.5097427997324202,
+        "regret_mae": 0.3921329892343945,
+        "success_accuracy": 0.8779226872656081,
+        "total_loss": 1.5752892427974277
+      }
+    },
+    {
+      "epoch": 24,
+      "train": {
+        "bc_loss": 0.08576599643048313,
+        "field_effect_loss": 0.009376069621389939,
+        "field_potential_loss": 0.22047014555169475,
+        "field_preference_loss": 0.49827172772751915,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18655536220305496,
+        "proposal_loss": 3.3681649066844934e-06,
+        "rank_acc": 0.8505271739429898,
+        "rank_loss": 0.498258162703779,
+        "regret_mae": 0.3540792941219277,
+        "success_accuracy": 0.8812954897681872,
+        "total_loss": 1.2353394561343722
+      },
+      "val": {
+        "bc_loss": 0.13938137392203012,
+        "field_effect_loss": 0.009277663219513165,
+        "field_potential_loss": 0.2653292194008827,
+        "field_preference_loss": 0.5137688567241033,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19404053688049316,
+        "proposal_loss": 2.8940887154425986e-06,
+        "rank_acc": 0.827929738495085,
+        "rank_loss": 0.5137633979320526,
+        "regret_mae": 0.39303411212232375,
+        "success_accuracy": 0.8753290209505293,
+        "total_loss": 1.5668853720029194
+      }
+    },
+    {
+      "epoch": 25,
+      "train": {
+        "bc_loss": 0.0814543502508766,
+        "field_effect_loss": 0.009363261355449341,
+        "field_potential_loss": 0.21932296889523664,
+        "field_preference_loss": 0.49654167683588135,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1871091986282004,
+        "proposal_loss": 3.17494489207901e-06,
+        "rank_acc": 0.8516269897421201,
+        "rank_loss": 0.496528048068285,
+        "regret_mae": 0.36070981952879166,
+        "success_accuracy": 0.8834113751848539,
+        "total_loss": 1.210647498567899
+      },
+      "val": {
+        "bc_loss": 0.14093191218045023,
+        "field_effect_loss": 0.009281607965628305,
+        "field_potential_loss": 0.2793622886141141,
+        "field_preference_loss": 0.5064755694733726,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19194430775112575,
+        "proposal_loss": 3.316251056478197e-06,
+        "rank_acc": 0.8285346262984805,
+        "rank_loss": 0.506469855705897,
+        "regret_mae": 0.4324636542134815,
+        "success_accuracy": 0.8704497085677253,
+        "total_loss": 1.5809441208839417
+      }
+    },
+    {
+      "epoch": 26,
+      "train": {
+        "bc_loss": 0.08190643171676332,
+        "field_effect_loss": 0.009374031344325177,
+        "field_potential_loss": 0.2209570827997393,
+        "field_preference_loss": 0.49740275823407704,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1877806948290931,
+        "proposal_loss": 3.235815954541168e-06,
+        "rank_acc": 0.8509750448995166,
+        "rank_loss": 0.4973853760295444,
+        "regret_mae": 0.3584645883076721,
+        "success_accuracy": 0.8806006279256609,
+        "total_loss": 1.2154834146300952
+      },
+      "val": {
+        "bc_loss": 0.1395697204603089,
+        "field_effect_loss": 0.009249651183684668,
+        "field_potential_loss": 0.2649960385428535,
+        "field_preference_loss": 0.5072847704092661,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1867915623717838,
+        "proposal_loss": 3.03697289736091e-06,
+        "rank_acc": 0.8329584466086494,
+        "rank_loss": 0.5072785168886185,
+        "regret_mae": 0.39045696788363987,
+        "success_accuracy": 0.8741144405470954,
+        "total_loss": 1.55989404519399
+      }
+    },
+    {
+      "epoch": 27,
+      "train": {
+        "bc_loss": 0.07931184199535185,
+        "field_effect_loss": 0.009342274706189832,
+        "field_potential_loss": 0.21837265168627104,
+        "field_preference_loss": 0.49662795745664173,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18560951844685608,
+        "proposal_loss": 3.1546752299820782e-06,
+        "rank_acc": 0.8519507961140739,
+        "rank_loss": 0.4966113418340683,
+        "regret_mae": 0.354548835919963,
+        "success_accuracy": 0.8807842549350526,
+        "total_loss": 1.1990551460120413
+      },
+      "val": {
+        "bc_loss": 0.1458885855972767,
+        "field_effect_loss": 0.009309433116060164,
+        "field_potential_loss": 0.2804822127024333,
+        "field_preference_loss": 0.5060126764906777,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18244297140174442,
+        "proposal_loss": 3.3602395130098253e-06,
+        "rank_acc": 0.8276396493117014,
+        "rank_loss": 0.5060083468755087,
+        "regret_mae": 0.390384445587794,
+        "success_accuracy": 0.8793647785981497,
+        "total_loss": 1.6050580210155911
+      }
+    },
+    {
+      "epoch": 28,
+      "train": {
+        "bc_loss": 0.07741976896714833,
+        "field_effect_loss": 0.009351225058910333,
+        "field_potential_loss": 0.21665144546164405,
+        "field_preference_loss": 0.49709269197450745,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18687419158716997,
+        "proposal_loss": 3.190585690365171e-06,
+        "rank_acc": 0.8522785007953644,
+        "rank_loss": 0.49707910377118325,
+        "regret_mae": 0.35349839698109364,
+        "success_accuracy": 0.882105118698544,
+        "total_loss": 1.18794261250231
+      },
+      "val": {
+        "bc_loss": 0.1440083367957009,
+        "field_effect_loss": 0.009252602519053552,
+        "field_potential_loss": 0.2710229837232166,
+        "field_preference_loss": 0.5107390367322497,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18460895203881794,
+        "proposal_loss": 3.080539651515816e-06,
+        "rank_acc": 0.8258205751578013,
+        "rank_loss": 0.5107326441340976,
+        "regret_mae": 0.37446049849192303,
+        "success_accuracy": 0.8718287944793701,
+        "total_loss": 1.5933859282069736
+      }
+    },
+    {
+      "epoch": 29,
+      "train": {
+        "bc_loss": 0.0802428335365322,
+        "field_effect_loss": 0.009307644517523132,
+        "field_potential_loss": 0.21315262549453312,
+        "field_preference_loss": 0.4941309309668011,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18591036254333126,
+        "proposal_loss": 3.033451633882578e-06,
+        "rank_acc": 0.8559404595030679,
+        "rank_loss": 0.4941122308373451,
+        "regret_mae": 0.3500535831683212,
+        "success_accuracy": 0.884799011879497,
+        "total_loss": 1.1949640446239047
+      },
+      "val": {
+        "bc_loss": 0.1388491830892033,
+        "field_effect_loss": 0.009226797407286035,
+        "field_potential_loss": 0.2551382887694571,
+        "field_preference_loss": 0.516539043850369,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1904102149936888,
+        "proposal_loss": 3.2502353709181384e-06,
+        "rank_acc": 0.8304381403658125,
+        "rank_loss": 0.516533311870363,
+        "regret_mae": 0.3661145286427604,
+        "success_accuracy": 0.8777371744314829,
+        "total_loss": 1.557318925857544
+      }
+    },
+    {
+      "epoch": 30,
+      "train": {
+        "bc_loss": 0.07759657947139607,
+        "field_effect_loss": 0.00929003096340845,
+        "field_potential_loss": 0.21027268717686334,
+        "field_preference_loss": 0.49346191560228664,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1858726197646724,
+        "proposal_loss": 3.1354000782357616e-06,
+        "rank_acc": 0.8551158408323923,
+        "rank_loss": 0.49344488936993813,
+        "regret_mae": 0.3489871794978778,
+        "success_accuracy": 0.8851266197032399,
+        "total_loss": 1.1779168388909764
+      },
+      "val": {
+        "bc_loss": 0.14676414016220304,
+        "field_effect_loss": 0.009254280296671722,
+        "field_potential_loss": 0.25933672736088437,
+        "field_preference_loss": 0.5165589037868712,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19169910997152328,
+        "proposal_loss": 3.246316497198778e-06,
+        "rank_acc": 0.8253436154789395,
+        "rank_loss": 0.5165523721112145,
+        "regret_mae": 0.36957623561223346,
+        "success_accuracy": 0.8751155071788363,
+        "total_loss": 1.6012506484985352
+      }
+    },
+    {
+      "epoch": 31,
+      "train": {
+        "bc_loss": 0.07483990826747483,
+        "field_effect_loss": 0.00927314810299625,
+        "field_potential_loss": 0.2077812287542555,
+        "field_preference_loss": 0.49253499549296165,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18684348298443687,
+        "proposal_loss": 3.255689714857605e-06,
+        "rank_acc": 0.8562688686781459,
+        "rank_loss": 0.4925190756718318,
+        "regret_mae": 0.3503772695031431,
+        "success_accuracy": 0.8836722092496024,
+        "total_loss": 1.1603098122609987
+      },
+      "val": {
+        "bc_loss": 0.14115538862016466,
+        "field_effect_loss": 0.009277352494084172,
+        "field_potential_loss": 0.2677536416384909,
+        "field_preference_loss": 0.5135424832503,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18503490090370178,
+        "proposal_loss": 2.9532186292474053e-06,
+        "rank_acc": 0.8275297151671516,
+        "rank_loss": 0.5135394881169001,
+        "regret_mae": 0.381787015332116,
+        "success_accuracy": 0.8833725353082021,
+        "total_loss": 1.5768415530522664
+      }
+    },
+    {
+      "epoch": 32,
+      "train": {
+        "bc_loss": 0.07662558358990484,
+        "field_effect_loss": 0.009275914299198322,
+        "field_potential_loss": 0.20679389064510664,
+        "field_preference_loss": 0.49246785044670105,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18443171038395828,
+        "proposal_loss": 2.9915527919557562e-06,
+        "rank_acc": 0.8578339964151382,
+        "rank_loss": 0.4924507008658515,
+        "regret_mae": 0.34085312195950085,
+        "success_accuracy": 0.886631110476123,
+        "total_loss": 1.1681191292073991
+      },
+      "val": {
+        "bc_loss": 0.14408120036953026,
+        "field_effect_loss": 0.009237894840124581,
+        "field_potential_loss": 0.2601384644707044,
+        "field_preference_loss": 0.5110623505380418,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18937111645936966,
+        "proposal_loss": 3.390953982792174e-06,
+        "rank_acc": 0.8312628931469388,
+        "rank_loss": 0.5110578305191464,
+        "regret_mae": 0.38793808387385476,
+        "success_accuracy": 0.8746674789322747,
+        "total_loss": 1.5821721620029874
+      }
+    },
+    {
+      "epoch": 33,
+      "train": {
+        "bc_loss": 0.0724950828589499,
+        "field_effect_loss": 0.00923701095679361,
+        "field_potential_loss": 0.20408037615319094,
+        "field_preference_loss": 0.49139439562956494,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1853681622693936,
+        "proposal_loss": 2.982490030944973e-06,
+        "rank_acc": 0.8585975476437144,
+        "rank_loss": 0.4913808550271723,
+        "regret_mae": 0.3472485972775353,
+        "success_accuracy": 0.8843941978282399,
+        "total_loss": 1.1432666521933343
+      },
+      "val": {
+        "bc_loss": 0.14606857051452002,
+        "field_effect_loss": 0.009258834205360876,
+        "field_potential_loss": 0.2637049812409613,
+        "field_preference_loss": 0.5174804147746828,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19151108298036787,
+        "proposal_loss": 3.0493338651164472e-06,
+        "rank_acc": 0.827435788181093,
+        "rank_loss": 0.5174748003482819,
+        "regret_mae": 0.4091388699081209,
+        "success_accuracy": 0.8740444382031759,
+        "total_loss": 1.6020700931549072
+      }
+    },
+    {
+      "epoch": 34,
+      "train": {
+        "bc_loss": 0.06825648328392869,
+        "field_effect_loss": 0.009256597375497222,
+        "field_potential_loss": 0.2039500849528445,
+        "field_preference_loss": 0.4908555696407954,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1843195812155803,
+        "proposal_loss": 3.0267992138356526e-06,
+        "rank_acc": 0.8593592743078867,
+        "rank_loss": 0.49083804173601997,
+        "regret_mae": 0.3511626674897141,
+        "success_accuracy": 0.8851808731754621,
+        "total_loss": 1.1215076661772199
+      },
+      "val": {
+        "bc_loss": 0.15108313577042687,
+        "field_effect_loss": 0.0092657549927632,
+        "field_potential_loss": 0.26973742494980496,
+        "field_preference_loss": 0.5062181585364871,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18580556412537894,
+        "proposal_loss": 2.6710457506043085e-06,
+        "rank_acc": 0.8309058712588416,
+        "rank_loss": 0.506217340628306,
+        "regret_mae": 0.397860093249215,
+        "success_accuracy": 0.8805513547526466,
+        "total_loss": 1.621411489115821
+      }
+    },
+    {
+      "epoch": 35,
+      "train": {
+        "bc_loss": 0.0689889055211097,
+        "field_effect_loss": 0.009237550334849706,
+        "field_potential_loss": 0.20071054436266422,
+        "field_preference_loss": 0.4900008589029312,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.185612045849363,
+        "proposal_loss": 3.022805366804783e-06,
+        "rank_acc": 0.861445851624012,
+        "rank_loss": 0.4899863422744804,
+        "regret_mae": 0.3473355062305927,
+        "success_accuracy": 0.8865539034207662,
+        "total_loss": 1.1205869010753102
+      },
+      "val": {
+        "bc_loss": 0.14639914656678835,
+        "field_effect_loss": 0.009308882807898853,
+        "field_potential_loss": 0.26579072574774426,
+        "field_preference_loss": 0.5201732797755135,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18895331935750115,
+        "proposal_loss": 2.8542722121629695e-06,
+        "rank_acc": 0.8268237312634786,
+        "rank_loss": 0.5201702449056838,
+        "regret_mae": 0.3750657836596171,
+        "success_accuracy": 0.8783882160981497,
+        "total_loss": 1.6102965142991807
+      }
+    },
+    {
+      "epoch": 36,
+      "train": {
+        "bc_loss": 0.06874692698733674,
+        "field_effect_loss": 0.009215531573216949,
+        "field_potential_loss": 0.19979151379730967,
+        "field_preference_loss": 0.49036448407504296,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18544859521918827,
+        "proposal_loss": 2.7992915679280686e-06,
+        "rank_acc": 0.8619132431017028,
+        "rank_loss": 0.49034315596024197,
+        "regret_mae": 0.33808669199546176,
+        "success_accuracy": 0.8847969257169299,
+        "total_loss": 1.1188483718368742
+      },
+      "val": {
+        "bc_loss": 0.1478497042424149,
+        "field_effect_loss": 0.009364152404790124,
+        "field_potential_loss": 0.27551161663399804,
+        "field_preference_loss": 0.5063382966650857,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18261539273791844,
+        "proposal_loss": 3.4695937706096124e-06,
+        "rank_acc": 0.828936861621009,
+        "rank_loss": 0.5063318941328261,
+        "regret_mae": 0.393838150633706,
+        "success_accuracy": 0.8808768755859799,
+        "total_loss": 1.6099106470743816
+      }
+    },
+    {
+      "epoch": 37,
+      "train": {
+        "bc_loss": 0.06955425544745392,
+        "field_effect_loss": 0.009240468963980675,
+        "field_potential_loss": 0.2019717451184988,
+        "field_preference_loss": 0.4905320753653844,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18467223913305336,
+        "proposal_loss": 2.8991978208776145e-06,
+        "rank_acc": 0.8598818878332773,
+        "rank_loss": 0.4905056245625019,
+        "regret_mae": 0.345250881380505,
+        "success_accuracy": 0.887100611295965,
+        "total_loss": 1.1252224726809397
+      },
+      "val": {
+        "bc_loss": 0.14545350273450217,
+        "field_effect_loss": 0.009268680147619711,
+        "field_potential_loss": 0.26310913927025265,
+        "field_preference_loss": 0.5155529148048825,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18994239883290398,
+        "proposal_loss": 2.8308371232318072e-06,
+        "rank_acc": 0.8311586115095351,
+        "rank_loss": 0.5155491332213084,
+        "regret_mae": 0.39120327764087254,
+        "success_accuracy": 0.8730538752343919,
+        "total_loss": 1.5979964137077332
+      }
+    },
+    {
+      "epoch": 38,
+      "train": {
+        "bc_loss": 0.06701447809528974,
+        "field_effect_loss": 0.009202092058128782,
+        "field_potential_loss": 0.19646307670821747,
+        "field_preference_loss": 0.4873090729945236,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1841707875331243,
+        "proposal_loss": 2.6974669664367766e-06,
+        "rank_acc": 0.8641413822770119,
+        "rank_loss": 0.4872872879107793,
+        "regret_mae": 0.3422290724184778,
+        "success_accuracy": 0.8857442744904094,
+        "total_loss": 1.1034042909741402
+      },
+      "val": {
+        "bc_loss": 0.14203310964836013,
+        "field_effect_loss": 0.009268712821520038,
+        "field_potential_loss": 0.26321512046787476,
+        "field_preference_loss": 0.5144331372446485,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1948101164566146,
+        "proposal_loss": 2.6159296642921013e-06,
+        "rank_acc": 0.8316035833623674,
+        "rank_loss": 0.5144275211625629,
+        "regret_mae": 0.39294591546058655,
+        "success_accuracy": 0.8707717292838626,
+        "total_loss": 1.5788562893867493
+      }
+    },
+    {
+      "epoch": 39,
+      "train": {
+        "bc_loss": 0.06537804592193829,
+        "field_effect_loss": 0.009135290411197476,
+        "field_potential_loss": 0.19242062533481252,
+        "field_preference_loss": 0.48693908875187236,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1855908510171705,
+        "proposal_loss": 2.9454996687642656e-06,
+        "rank_acc": 0.8645853209826682,
+        "rank_loss": 0.48691798746585846,
+        "regret_mae": 0.34609640141328174,
+        "success_accuracy": 0.8862763759162691,
+        "total_loss": 1.0901164967152808
+      },
+      "val": {
+        "bc_loss": 0.14972315438919598,
+        "field_effect_loss": 0.009233337852896916,
+        "field_potential_loss": 0.27685275425513584,
+        "field_preference_loss": 0.5060797118478351,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18568002929290137,
+        "proposal_loss": 2.9693663666168705e-06,
+        "rank_acc": 0.8292485475540161,
+        "rank_loss": 0.5060749368535148,
+        "regret_mae": 0.41037705209520126,
+        "success_accuracy": 0.8772716455989413,
+        "total_loss": 1.6202431850963168
+      }
+    },
+    {
+      "epoch": 40,
+      "train": {
+        "bc_loss": 0.06500557763502002,
+        "field_effect_loss": 0.009115403912599303,
+        "field_potential_loss": 0.18964361689156956,
+        "field_preference_loss": 0.48682501456803745,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18445953664680323,
+        "proposal_loss": 3.0262607171809777e-06,
+        "rank_acc": 0.8650213537944688,
+        "rank_loss": 0.48681171528167194,
+        "regret_mae": 0.33955201341046226,
+        "success_accuracy": 0.8861073553562164,
+        "total_loss": 1.085644477771388
+      },
+      "val": {
+        "bc_loss": 0.14890781541665396,
+        "field_effect_loss": 0.00929537353416284,
+        "field_potential_loss": 0.2713838318983714,
+        "field_preference_loss": 0.509952598147922,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18912295914358562,
+        "proposal_loss": 3.021175391242852e-06,
+        "rank_acc": 0.8287045194043053,
+        "rank_loss": 0.5099461310439639,
+        "regret_mae": 0.393127355310652,
+        "success_accuracy": 0.8775061600738101,
+        "total_loss": 1.615245799223582
+      }
+    },
+    {
+      "epoch": 41,
+      "train": {
+        "bc_loss": 0.06293584789252943,
+        "field_effect_loss": 0.009134675537805177,
+        "field_potential_loss": 0.19135429710149765,
+        "field_preference_loss": 0.48594580094019574,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18364913885792097,
+        "proposal_loss": 2.7789885797978463e-06,
+        "rank_acc": 0.8673899852567248,
+        "rank_loss": 0.4859270416200161,
+        "regret_mae": 0.33673925201098126,
+        "success_accuracy": 0.88666449735562,
+        "total_loss": 1.0760895469122462
+      },
+      "val": {
+        "bc_loss": 0.14833149976200527,
+        "field_effect_loss": 0.009410339304142527,
+        "field_potential_loss": 0.2935059832202064,
+        "field_preference_loss": 0.49914594491322833,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18534359087546667,
+        "proposal_loss": 3.343468380181649e-06,
+        "rank_acc": 0.8285814192559984,
+        "rank_loss": 0.4991397476858563,
+        "regret_mae": 0.4367992613050673,
+        "success_accuracy": 0.8799003130859799,
+        "total_loss": 1.6230118208461337
+      }
+    },
+    {
+      "epoch": 42,
+      "train": {
+        "bc_loss": 0.06402667777405845,
+        "field_effect_loss": 0.009098148710715273,
+        "field_potential_loss": 0.18682758986122078,
+        "field_preference_loss": 0.4842464046345817,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1842279423856073,
+        "proposal_loss": 2.9227203444608877e-06,
+        "rank_acc": 0.8686396554112434,
+        "rank_loss": 0.48423215995232266,
+        "regret_mae": 0.3438873543507523,
+        "success_accuracy": 0.8872612855500646,
+        "total_loss": 1.0744035070141156
+      },
+      "val": {
+        "bc_loss": 0.15734340457452667,
+        "field_effect_loss": 0.009331342815938923,
+        "field_potential_loss": 0.27379757165908813,
+        "field_preference_loss": 0.513781319061915,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18685181025001737,
+        "proposal_loss": 3.013388171287564e-06,
+        "rank_acc": 0.825360377629598,
+        "rank_loss": 0.5137757923867967,
+        "regret_mae": 0.40055159562163883,
+        "success_accuracy": 0.877733674314287,
+        "total_loss": 1.6644804279009502
+      }
+    },
+    {
+      "epoch": 43,
+      "train": {
+        "bc_loss": 0.06295879278331995,
+        "field_effect_loss": 0.00909574006477164,
+        "field_potential_loss": 0.18578340920309225,
+        "field_preference_loss": 0.4842660319474008,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18361005393995178,
+        "proposal_loss": 2.7568564493978756e-06,
+        "rank_acc": 0.8691810982094871,
+        "rank_loss": 0.48424215490619343,
+        "regret_mae": 0.34374551226695377,
+        "success_accuracy": 0.889099643462234,
+        "total_loss": 1.0679754656222131
+      },
+      "val": {
+        "bc_loss": 0.14956712391641405,
+        "field_effect_loss": 0.009357012989413407,
+        "field_potential_loss": 0.285794686112139,
+        "field_preference_loss": 0.5048026012049781,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18432870672808754,
+        "proposal_loss": 2.8084066722537195e-06,
+        "rank_acc": 0.8277391195297241,
+        "rank_loss": 0.5047980944315592,
+        "regret_mae": 0.40559210545486873,
+        "success_accuracy": 0.8780626952648163,
+        "total_loss": 1.6284161276287503
+      }
+    },
+    {
+      "epoch": 44,
+      "train": {
+        "bc_loss": 0.061444741828987993,
+        "field_effect_loss": 0.00905682844394404,
+        "field_potential_loss": 0.18113447073847055,
+        "field_preference_loss": 0.48280205329259235,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18258334054715103,
+        "proposal_loss": 2.558752467191678e-06,
+        "rank_acc": 0.869562485979663,
+        "rank_loss": 0.48278311805592644,
+        "regret_mae": 0.3372274976637628,
+        "success_accuracy": 0.8887887290782399,
+        "total_loss": 1.0538840318719547
+      },
+      "val": {
+        "bc_loss": 0.14728901121351454,
+        "field_effect_loss": 0.009354393204881085,
+        "field_potential_loss": 0.280954553021325,
+        "field_preference_loss": 0.5068592263592614,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18672911077737808,
+        "proposal_loss": 2.343760734725543e-06,
+        "rank_acc": 0.8289147681660123,
+        "rank_loss": 0.5068550109863281,
+        "regret_mae": 0.3859325349330902,
+        "success_accuracy": 0.8808873759375678,
+        "total_loss": 1.6147269010543823
+      }
+    },
+    {
+      "epoch": 45,
+      "train": {
+        "bc_loss": 0.05951930209994316,
+        "field_effect_loss": 0.009032687249903878,
+        "field_potential_loss": 0.17834003207584223,
+        "field_preference_loss": 0.48129212152626777,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18356457435422474,
+        "proposal_loss": 2.494914116684615e-06,
+        "rank_acc": 0.8712582737207413,
+        "rank_loss": 0.481260198685858,
+        "regret_mae": 0.3385179963790708,
+        "success_accuracy": 0.8894543771942457,
+        "total_loss": 1.039552304479811
+      },
+      "val": {
+        "bc_loss": 0.1464761176870929,
+        "field_effect_loss": 0.009371131503333649,
+        "field_potential_loss": 0.28035490959882736,
+        "field_preference_loss": 0.5151016496949725,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18878484186198977,
+        "proposal_loss": 2.3431966206367683e-06,
+        "rank_acc": 0.8214039968119727,
+        "rank_loss": 0.5150964458783468,
+        "regret_mae": 0.41932618452443016,
+        "success_accuracy": 0.8781467013888888,
+        "total_loss": 1.6167868839369879
+      }
+    },
+    {
+      "epoch": 46,
+      "train": {
+        "bc_loss": 0.05926642018473811,
+        "field_effect_loss": 0.009031233093183901,
+        "field_potential_loss": 0.17749296315014362,
+        "field_preference_loss": 0.48113323872288066,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18351849127146933,
+        "proposal_loss": 2.4718687099165336e-06,
+        "rank_acc": 0.8720171360505952,
+        "rank_loss": 0.4811138076086839,
+        "regret_mae": 0.3371584328512351,
+        "success_accuracy": 0.889437683754497,
+        "total_loss": 1.0375731868876352
+      },
+      "val": {
+        "bc_loss": 0.14808820022477043,
+        "field_effect_loss": 0.009363930025655363,
+        "field_potential_loss": 0.28279872486988705,
+        "field_preference_loss": 0.5117519398530325,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18514027612076867,
+        "proposal_loss": 2.159975338145159e-06,
+        "rank_acc": 0.824683944384257,
+        "rank_loss": 0.5117466068930097,
+        "regret_mae": 0.41205907695823246,
+        "success_accuracy": 0.8799038132031759,
+        "total_loss": 1.6238711675008137
+      }
+    },
+    {
+      "epoch": 47,
+      "train": {
+        "bc_loss": 0.05694263500885831,
+        "field_effect_loss": 0.009006609299427105,
+        "field_potential_loss": 0.17553690448403358,
+        "field_preference_loss": 0.48142344049281544,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1822759242107471,
+        "proposal_loss": 2.6290596698193744e-06,
+        "rank_acc": 0.872896229227384,
+        "rank_loss": 0.48139970873792964,
+        "regret_mae": 0.33778492071562344,
+        "success_accuracy": 0.8906729941566786,
+        "total_loss": 1.0237259682681825
+      },
+      "val": {
+        "bc_loss": 0.1462397254589531,
+        "field_effect_loss": 0.009395269599432746,
+        "field_potential_loss": 0.28438176297479206,
+        "field_preference_loss": 0.5092861933840646,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1836638632747862,
+        "proposal_loss": 2.6982413222261434e-06,
+        "rank_acc": 0.8235262533028921,
+        "rank_loss": 0.5092802660332786,
+        "regret_mae": 0.3844190670384301,
+        "success_accuracy": 0.8783777157465616,
+        "total_loss": 1.6157546043395996
+      }
+    },
+    {
+      "epoch": 48,
+      "train": {
+        "bc_loss": 0.05878396916927563,
+        "field_effect_loss": 0.008994072125965936,
+        "field_potential_loss": 0.17520549893379211,
+        "field_preference_loss": 0.48048442064060104,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18229366652667522,
+        "proposal_loss": 2.5363087938027698e-06,
+        "rank_acc": 0.8733426754673322,
+        "rank_loss": 0.48046915150350994,
+        "regret_mae": 0.3422783373130692,
+        "success_accuracy": 0.891620342930158,
+        "total_loss": 1.0316113175617323
+      },
+      "val": {
+        "bc_loss": 0.15010524582531717,
+        "field_effect_loss": 0.00934282667003572,
+        "field_potential_loss": 0.2831971123814583,
+        "field_preference_loss": 0.5071640031205283,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18438772443268034,
+        "proposal_loss": 2.7799950683200022e-06,
+        "rank_acc": 0.8268875314129723,
+        "rank_loss": 0.5071600162320666,
+        "regret_mae": 0.4141058474779129,
+        "success_accuracy": 0.8788152436415354,
+        "total_loss": 1.6306420101059809
+      }
+    },
+    {
+      "epoch": 49,
+      "train": {
+        "bc_loss": 0.05797625439138048,
+        "field_effect_loss": 0.008986926572914753,
+        "field_potential_loss": 0.1731946204478542,
+        "field_preference_loss": 0.47809188647402656,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1814902006751961,
+        "proposal_loss": 2.443910970681726e-06,
+        "rank_acc": 0.8755473055773311,
+        "rank_loss": 0.47807611856195664,
+        "regret_mae": 0.3381124378906356,
+        "success_accuracy": 0.892346504661772,
+        "total_loss": 1.0229609914951854
+      },
+      "val": {
+        "bc_loss": 0.1509255278441641,
+        "field_effect_loss": 0.009275097011898955,
+        "field_potential_loss": 0.2687208677331607,
+        "field_preference_loss": 0.5161103970474668,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1931292778915829,
+        "proposal_loss": 3.0542429006648795e-06,
+        "rank_acc": 0.825303402211931,
+        "rank_loss": 0.5161049067974091,
+        "regret_mae": 0.39289865063296425,
+        "success_accuracy": 0.8755530350738101,
+        "total_loss": 1.6304601695802476
+      }
+    },
+    {
+      "epoch": 50,
+      "train": {
+        "bc_loss": 0.05701940558436844,
+        "field_effect_loss": 0.008950961656713238,
+        "field_potential_loss": 0.168738115992811,
+        "field_preference_loss": 0.4786707129743364,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18191195920937592,
+        "proposal_loss": 2.5393535515326625e-06,
+        "rank_acc": 0.8751111138198111,
+        "rank_loss": 0.4786437770558728,
+        "regret_mae": 0.3377168441398276,
+        "success_accuracy": 0.8924028451244036,
+        "total_loss": 1.0139043380816777
+      },
+      "val": {
+        "bc_loss": 0.1473568015628391,
+        "field_effect_loss": 0.009332833190759024,
+        "field_potential_loss": 0.2772196051147249,
+        "field_preference_loss": 0.5135047800011106,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1909649603896671,
+        "proposal_loss": 2.611224292397512e-06,
+        "rank_acc": 0.8269278572665321,
+        "rank_loss": 0.5135002649492688,
+        "regret_mae": 0.4162008050415251,
+        "success_accuracy": 0.8759905629687839,
+        "total_loss": 1.617375910282135
+      }
+    }
+  ]
+}

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_0/resolved_config.json

@@ -0,0 +1,69 @@
+{
+  "action_dim": 7,
+  "action_horizon": 16,
+  "backbone_feature_batch_size": 64,
+  "backbone_feature_cache": null,
+  "backbone_freeze": true,
+  "backbone_local_files_only": true,
+  "backbone_model": null,
+  "backbone_type": "native",
+  "batch_groups": 32,
+  "batch_size_groups": null,
+  "dataset_dir": "/scratch/knguy52/dovla/experiments/h16_merged_dataset",
+  "device": "cuda",
+  "effect_dim": 32,
+  "epochs": 50,
+  "hidden_dim": 256,
+  "lang_dim": 64,
+  "lattice_neighbors": 32,
+  "learning_rate": 0.001,
+  "losses": {
+    "bc": 5.0,
+    "bc_best_action": 1.0,
+    "causal_contrastive": 0.5,
+    "contrast": 0.5,
+    "effect": 1.0,
+    "field_anchor": 0.25,
+    "field_effect": 1.0,
+    "field_potential": 1.0,
+    "field_preference": 1.0,
+    "forward_effect_prediction": 1.0,
+    "lang_pair": 0.25,
+    "language_minimal_pair": 0.25,
+    "progress": 1.0,
+    "proposal": 2.0,
+    "rank": 1.0,
+    "regret": 0.5,
+    "regret_prediction": 0.5,
+    "same_state_pairwise_ranking": 1.0,
+    "success": 1.0
+  },
+  "lr": null,
+  "objective": "lattice_field",
+  "obs_dim": 96,
+  "observation_mode": "state",
+  "output_dir": "/scratch/knguy52/dovla/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_0",
+  "pair_count_per_group": 32,
+  "pair_scope": "same_state",
+  "policy_target_map": null,
+  "policy_target_map_coverage": {
+    "num_targets": 0,
+    "train_coverage": 0.0,
+    "train_covered": 0,
+    "train_groups": 2298,
+    "val_coverage": 0.0,
+    "val_covered": 0,
+    "val_groups": 575
+  },
+  "policy_target_types": [
+    "near_miss"
+  ],
+  "proposal_types": [
+    "no_op"
+  ],
+  "records_per_group": 16,
+  "seed": 0,
+  "val_fraction": 0.2,
+  "wandb": false,
+  "weight_decay": 0.0
+}

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_1/best.pt

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_1/best_policy.pt

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_1/latest.pt

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_1/metrics.json

@@ -0,0 +1,1582 @@
+{
+  "best": {
+    "bc_loss": 0.13860523245400852,
+    "field_effect_loss": 0.01002420966203014,
+    "field_potential_loss": 0.2973766442802217,
+    "field_preference_loss": 0.5044528775744967,
+    "lattice_edges": 3833.3333333333335,
+    "progress_mae": 0.19975476380851534,
+    "proposal_loss": 2.9795605390366595e-06,
+    "rank_acc": 0.8332956035931905,
+    "rank_loss": 0.504441499710083,
+    "regret_mae": 0.427681823571523,
+    "success_accuracy": 0.8721823195616404,
+    "total_loss": 1.5891507400406732
+  },
+  "best_policy": {
+    "bc_loss": 0.1322652217414644,
+    "field_effect_loss": 0.010026000388380554,
+    "field_potential_loss": 0.2885795517100228,
+    "field_preference_loss": 0.5062332368559308,
+    "lattice_edges": 3833.3333333333335,
+    "progress_mae": 0.1950002759695053,
+    "proposal_loss": 3.3075696895442284e-06,
+    "rank_acc": 0.8296620117293464,
+    "rank_loss": 0.5062223523855209,
+    "regret_mae": 0.3615676628218757,
+    "success_accuracy": 0.8715242776605818,
+    "total_loss": 1.5516093638208177
+  },
+  "history": [
+    {
+      "epoch": 1,
+      "train": {
+        "bc_loss": 0.31296772323548794,
+        "field_effect_loss": 0.011810641891012589,
+        "field_potential_loss": 0.36331021702951855,
+        "field_preference_loss": 0.5551957624653975,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.20024140593078402,
+        "proposal_loss": 0.0007920488194738735,
+        "rank_acc": 0.7690677617986997,
+        "rank_loss": 0.5551961871484915,
+        "regret_mae": 0.6211433977716498,
+        "success_accuracy": 0.8660606973701053,
+        "total_loss": 2.6001286258300147
+      },
+      "val": {
+        "bc_loss": 0.219724769393603,
+        "field_effect_loss": 0.011381767741921876,
+        "field_potential_loss": 0.35945942997932434,
+        "field_preference_loss": 0.5411418742603726,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20132794893450207,
+        "proposal_loss": 0.00011022848209702513,
+        "rank_acc": 0.7892049352327982,
+        "rank_loss": 0.5411339534653558,
+        "regret_mae": 0.5675967832406362,
+        "success_accuracy": 0.8528120782640245,
+        "total_loss": 2.1136290497250028
+      }
+    },
+    {
+      "epoch": 2,
+      "train": {
+        "bc_loss": 0.20893590038435328,
+        "field_effect_loss": 0.010620662024141185,
+        "field_potential_loss": 0.3267577149801784,
+        "field_preference_loss": 0.5358565383487277,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1904634918189711,
+        "proposal_loss": 8.283138984350242e-05,
+        "rank_acc": 0.7988820349176725,
+        "rank_loss": 0.5358563305603133,
+        "regret_mae": 0.5414891271955438,
+        "success_accuracy": 0.8666303588284386,
+        "total_loss": 2.0116010109583535
+      },
+      "val": {
+        "bc_loss": 0.19223365725742447,
+        "field_effect_loss": 0.01092043218927251,
+        "field_potential_loss": 0.3660995049609078,
+        "field_preference_loss": 0.517738820777999,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19869421670834223,
+        "proposal_loss": 5.327659669597374e-05,
+        "rank_acc": 0.7973560260401832,
+        "rank_loss": 0.5177260637283325,
+        "regret_mae": 0.45316366685761345,
+        "success_accuracy": 0.855853772825665,
+        "total_loss": 1.951379981305864
+      }
+    },
+    {
+      "epoch": 3,
+      "train": {
+        "bc_loss": 0.18724164573682678,
+        "field_effect_loss": 0.010248898166335292,
+        "field_potential_loss": 0.29611779438952607,
+        "field_preference_loss": 0.5274873471094502,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18994024851255947,
+        "proposal_loss": 4.489911743298257e-05,
+        "rank_acc": 0.8057426876491971,
+        "rank_loss": 0.5274814338319831,
+        "regret_mae": 0.459577024810844,
+        "success_accuracy": 0.8700274609857135,
+        "total_loss": 1.859152239229944
+      },
+      "val": {
+        "bc_loss": 0.17215145669049686,
+        "field_effect_loss": 0.010834043328132894,
+        "field_potential_loss": 0.3184017241001129,
+        "field_preference_loss": 0.5455676118532816,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.21410448104143143,
+        "proposal_loss": 3.390187268653083e-05,
+        "rank_acc": 0.8041569093863169,
+        "rank_loss": 0.5455550816324022,
+        "regret_mae": 0.5286538518137402,
+        "success_accuracy": 0.8543241752518548,
+        "total_loss": 1.8343389564090304
+      }
+    },
+    {
+      "epoch": 4,
+      "train": {
+        "bc_loss": 0.17238479842328364,
+        "field_effect_loss": 0.010112364155550798,
+        "field_potential_loss": 0.28761785539488,
+        "field_preference_loss": 0.5234924157460531,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1894432289732827,
+        "proposal_loss": 2.854011838988097e-05,
+        "rank_acc": 0.8121899093190829,
+        "rank_loss": 0.5234895700381862,
+        "regret_mae": 0.448579808904065,
+        "success_accuracy": 0.8707327561246024,
+        "total_loss": 1.7712572283214993
+      },
+      "val": {
+        "bc_loss": 0.1717616460389561,
+        "field_effect_loss": 0.010587147592256466,
+        "field_potential_loss": 0.30925074385272133,
+        "field_preference_loss": 0.5437865720854865,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20189093467262056,
+        "proposal_loss": 2.3317615462777514e-05,
+        "rank_acc": 0.802943444914288,
+        "rank_loss": 0.5437757240401374,
+        "regret_mae": 0.5117809342013465,
+        "success_accuracy": 0.8528015779124366,
+        "total_loss": 1.8220204247368708
+      }
+    },
+    {
+      "epoch": 5,
+      "train": {
+        "bc_loss": 0.15952119479576746,
+        "field_effect_loss": 0.010005261891314553,
+        "field_potential_loss": 0.2789178585840596,
+        "field_preference_loss": 0.5222767641146978,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1894255977951818,
+        "proposal_loss": 2.1056113218441573e-05,
+        "rank_acc": 0.8162725509868728,
+        "rank_loss": 0.5222692502041658,
+        "regret_mae": 0.44478662808736164,
+        "success_accuracy": 0.8723061068190469,
+        "total_loss": 1.6957630448871188
+      },
+      "val": {
+        "bc_loss": 0.16199031679166687,
+        "field_effect_loss": 0.010518429283466604,
+        "field_potential_loss": 0.2982821406589614,
+        "field_preference_loss": 0.5284048616886139,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20057249234782326,
+        "proposal_loss": 1.873741429234441e-05,
+        "rank_acc": 0.8084833953115675,
+        "rank_loss": 0.5283947752581702,
+        "regret_mae": 0.42593537271022797,
+        "success_accuracy": 0.8594205015235477,
+        "total_loss": 1.7413400411605835
+      }
+    },
+    {
+      "epoch": 6,
+      "train": {
+        "bc_loss": 0.15513312071561813,
+        "field_effect_loss": 0.009846322962807285,
+        "field_potential_loss": 0.26337927186654675,
+        "field_preference_loss": 0.5171056894792451,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1892595226979918,
+        "proposal_loss": 1.5970483382766866e-05,
+        "rank_acc": 0.820447736316257,
+        "rank_loss": 0.5170957951082124,
+        "regret_mae": 0.41232535822523964,
+        "success_accuracy": 0.8763271239068773,
+        "total_loss": 1.6504632532596588
+      },
+      "val": {
+        "bc_loss": 0.151757024642494,
+        "field_effect_loss": 0.010355569473985169,
+        "field_potential_loss": 0.29752831078237957,
+        "field_preference_loss": 0.5243177082803514,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.2044004367457496,
+        "proposal_loss": 1.393995100645245e-05,
+        "rank_acc": 0.811343080467648,
+        "rank_loss": 0.5243046681086222,
+        "regret_mae": 0.43599731226762134,
+        "success_accuracy": 0.863655772474077,
+        "total_loss": 1.6819879081514146
+      }
+    },
+    {
+      "epoch": 7,
+      "train": {
+        "bc_loss": 0.14436460234638718,
+        "field_effect_loss": 0.00982099230815139,
+        "field_potential_loss": 0.26482414288653267,
+        "field_preference_loss": 0.5161220936311616,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18709820922878054,
+        "proposal_loss": 1.301600678339633e-05,
+        "rank_acc": 0.8228970749510659,
+        "rank_loss": 0.5161164386404885,
+        "regret_mae": 0.4164523184299469,
+        "success_accuracy": 0.8784701360596551,
+        "total_loss": 1.597039928038915
+      },
+      "val": {
+        "bc_loss": 0.15123214738236535,
+        "field_effect_loss": 0.01031327454580201,
+        "field_potential_loss": 0.2811409913831287,
+        "field_preference_loss": 0.541976574394438,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.21122217261128956,
+        "proposal_loss": 1.1231953825093418e-05,
+        "rank_acc": 0.815445085366567,
+        "rank_loss": 0.5419691238138411,
+        "regret_mae": 0.43520354562335545,
+        "success_accuracy": 0.8604040642579397,
+        "total_loss": 1.6837305890189276
+      }
+    },
+    {
+      "epoch": 8,
+      "train": {
+        "bc_loss": 0.14031092098189724,
+        "field_effect_loss": 0.00978723000217643,
+        "field_potential_loss": 0.26264756359159946,
+        "field_preference_loss": 0.5157175896068414,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18933316465053293,
+        "proposal_loss": 9.643780736546938e-06,
+        "rank_acc": 0.8254914730787277,
+        "rank_loss": 0.5157074141833518,
+        "regret_mae": 0.40910257026553154,
+        "success_accuracy": 0.8745200658837954,
+        "total_loss": 1.574245548910565
+      },
+      "val": {
+        "bc_loss": 0.14860587442914644,
+        "field_effect_loss": 0.010304437691552771,
+        "field_potential_loss": 0.31997156474325394,
+        "field_preference_loss": 0.5090950462553236,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19349187364180884,
+        "proposal_loss": 8.61309149513545e-06,
+        "rank_acc": 0.8137714333004422,
+        "rank_loss": 0.5090838786628511,
+        "regret_mae": 0.37783128519852954,
+        "success_accuracy": 0.8612651195791032,
+        "total_loss": 1.672467119163937
+      }
+    },
+    {
+      "epoch": 9,
+      "train": {
+        "bc_loss": 0.13242660297287834,
+        "field_effect_loss": 0.009684302850574669,
+        "field_potential_loss": 0.2538694998042451,
+        "field_preference_loss": 0.5135682597756386,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18781670307119688,
+        "proposal_loss": 7.656315820743961e-06,
+        "rank_acc": 0.8288246393203735,
+        "rank_loss": 0.5135587342083454,
+        "regret_mae": 0.39859839984112316,
+        "success_accuracy": 0.8762311372492049,
+        "total_loss": 1.5225901272561815
+      },
+      "val": {
+        "bc_loss": 0.14214949061473212,
+        "field_effect_loss": 0.010262287066628536,
+        "field_potential_loss": 0.30470698409610325,
+        "field_preference_loss": 0.5165494266483519,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19556173019939,
+        "proposal_loss": 6.9081727360753575e-06,
+        "rank_acc": 0.8133239381843143,
+        "rank_loss": 0.5165384842289819,
+        "regret_mae": 0.3870216922627555,
+        "success_accuracy": 0.8682130641407437,
+        "total_loss": 1.6311715973748102
+      }
+    },
+    {
+      "epoch": 10,
+      "train": {
+        "bc_loss": 0.13157107391291195,
+        "field_effect_loss": 0.00962610837046264,
+        "field_potential_loss": 0.24829600068430105,
+        "field_preference_loss": 0.5117376703355048,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18767704628407955,
+        "proposal_loss": 6.738337813456181e-06,
+        "rank_acc": 0.8302827576796213,
+        "rank_loss": 0.5117301365567578,
+        "regret_mae": 0.3866368904709816,
+        "success_accuracy": 0.8804023944669299,
+        "total_loss": 1.5099299914307065
+      },
+      "val": {
+        "bc_loss": 0.14464630269342,
+        "field_effect_loss": 0.010193408725576268,
+        "field_potential_loss": 0.29386471377478707,
+        "field_preference_loss": 0.5175793849759631,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19899089386065802,
+        "proposal_loss": 6.061432815032377e-06,
+        "rank_acc": 0.818389183945126,
+        "rank_loss": 0.5175664126873016,
+        "regret_mae": 0.39913591080241734,
+        "success_accuracy": 0.8649578558074104,
+        "total_loss": 1.635732114315033
+      }
+    },
+    {
+      "epoch": 11,
+      "train": {
+        "bc_loss": 0.1251100299672948,
+        "field_effect_loss": 0.009571528980611928,
+        "field_potential_loss": 0.2454149251182874,
+        "field_preference_loss": 0.5106041555603346,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18836527028017575,
+        "proposal_loss": 6.002791426201131e-06,
+        "rank_acc": 0.833587307896879,
+        "rank_loss": 0.510593929224544,
+        "regret_mae": 0.39070795186691815,
+        "success_accuracy": 0.8772410858008597,
+        "total_loss": 1.4732304612795513
+      },
+      "val": {
+        "bc_loss": 0.1423600117365519,
+        "field_effect_loss": 0.01011422411021259,
+        "field_potential_loss": 0.2887183328469594,
+        "field_preference_loss": 0.5187903907563951,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20060071018007067,
+        "proposal_loss": 5.255824337129727e-06,
+        "rank_acc": 0.8217059108946059,
+        "rank_loss": 0.51877846982744,
+        "regret_mae": 0.3740177121427324,
+        "success_accuracy": 0.8662669393751357,
+        "total_loss": 1.618718061182234
+      }
+    },
+    {
+      "epoch": 12,
+      "train": {
+        "bc_loss": 0.12037536315619946,
+        "field_effect_loss": 0.009530804132939212,
+        "field_potential_loss": 0.24360166055460772,
+        "field_preference_loss": 0.5092900527848138,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18903819492293727,
+        "proposal_loss": 5.493830195114748e-06,
+        "rank_acc": 0.8348441653781467,
+        "rank_loss": 0.5092777026196321,
+        "regret_mae": 0.3862766996026039,
+        "success_accuracy": 0.8778190943929884,
+        "total_loss": 1.4463665319813623
+      },
+      "val": {
+        "bc_loss": 0.14104630053043365,
+        "field_effect_loss": 0.01015661625812451,
+        "field_potential_loss": 0.2768987715244293,
+        "field_preference_loss": 0.5311233699321747,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20527827988068262,
+        "proposal_loss": 5.099688299499879e-06,
+        "rank_acc": 0.8170826931794485,
+        "rank_loss": 0.5311112403869629,
+        "regret_mae": 0.40441307259930503,
+        "success_accuracy": 0.8603970640235477,
+        "total_loss": 1.6169270541932848
+      }
+    },
+    {
+      "epoch": 13,
+      "train": {
+        "bc_loss": 0.11695418330944246,
+        "field_effect_loss": 0.009511143214897148,
+        "field_potential_loss": 0.24320428756376108,
+        "field_preference_loss": 0.5091210533347394,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18761826078924868,
+        "proposal_loss": 4.906123880093421e-06,
+        "rank_acc": 0.8361889926923646,
+        "rank_loss": 0.5091118899484476,
+        "regret_mae": 0.37988723980055916,
+        "success_accuracy": 0.878180088268386,
+        "total_loss": 1.4284029172526465
+      },
+      "val": {
+        "bc_loss": 0.13601411382357279,
+        "field_effect_loss": 0.010079491831776168,
+        "field_potential_loss": 0.2896474964088864,
+        "field_preference_loss": 0.5199059579107497,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19661901311741936,
+        "proposal_loss": 4.5170224009578105e-06,
+        "rank_acc": 0.8222160206900703,
+        "rank_loss": 0.5198971331119537,
+        "regret_mae": 0.35909534990787506,
+        "success_accuracy": 0.8661549323134952,
+        "total_loss": 1.5874425768852234
+      }
+    },
+    {
+      "epoch": 14,
+      "train": {
+        "bc_loss": 0.10912573296162817,
+        "field_effect_loss": 0.009441794601217326,
+        "field_potential_loss": 0.23623204231262207,
+        "field_preference_loss": 0.5073767031232516,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18813862134185103,
+        "proposal_loss": 4.541388452455269e-06,
+        "rank_acc": 0.8385562259289954,
+        "rank_loss": 0.5073651601043012,
+        "regret_mae": 0.37585288865698707,
+        "success_accuracy": 0.8780027214023802,
+        "total_loss": 1.3796076393789716
+      },
+      "val": {
+        "bc_loss": 0.1363932519323296,
+        "field_effect_loss": 0.010101750544789765,
+        "field_potential_loss": 0.3036942366096709,
+        "field_preference_loss": 0.5067911495765051,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1935932007100847,
+        "proposal_loss": 4.532497162775447e-06,
+        "rank_acc": 0.8264922863907285,
+        "rank_loss": 0.5067797833018832,
+        "regret_mae": 0.384041138821178,
+        "success_accuracy": 0.8653883834679922,
+        "total_loss": 1.5897831453217401
+      }
+    },
+    {
+      "epoch": 15,
+      "train": {
+        "bc_loss": 0.10830302831406395,
+        "field_effect_loss": 0.00943811125923983,
+        "field_potential_loss": 0.23716292012896803,
+        "field_preference_loss": 0.5062191002070904,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18776153110795551,
+        "proposal_loss": 4.330689083392119e-06,
+        "rank_acc": 0.8386877038412623,
+        "rank_loss": 0.506208417730199,
+        "regret_mae": 0.3715030066668987,
+        "success_accuracy": 0.8776208601064153,
+        "total_loss": 1.375307901038064
+      },
+      "val": {
+        "bc_loss": 0.13442131421632236,
+        "field_effect_loss": 0.010174330479154984,
+        "field_potential_loss": 0.30553974873489803,
+        "field_preference_loss": 0.5084694094128079,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19356300019555622,
+        "proposal_loss": 3.843540967965156e-06,
+        "rank_acc": 0.8218472533755832,
+        "rank_loss": 0.5084589802556567,
+        "regret_mae": 0.3913183742099338,
+        "success_accuracy": 0.8707122239801619,
+        "total_loss": 1.584278576903873
+      }
+    },
+    {
+      "epoch": 16,
+      "train": {
+        "bc_loss": 0.10188614101045662,
+        "field_effect_loss": 0.009414009361838302,
+        "field_potential_loss": 0.2328791038857566,
+        "field_preference_loss": 0.5056630066699452,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18803729861974716,
+        "proposal_loss": 4.132846059848008e-06,
+        "rank_acc": 0.8421910670068529,
+        "rank_loss": 0.5056520613531271,
+        "regret_mae": 0.3702433125840293,
+        "success_accuracy": 0.8801999870273802,
+        "total_loss": 1.3378891497850418
+      },
+      "val": {
+        "bc_loss": 0.1420320611861017,
+        "field_effect_loss": 0.010014857579436567,
+        "field_potential_loss": 0.2841916357477506,
+        "field_preference_loss": 0.5152987589438757,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19920555336607826,
+        "proposal_loss": 3.669920374704816e-06,
+        "rank_acc": 0.8237650891145071,
+        "rank_loss": 0.5152874539295832,
+        "regret_mae": 0.36203480429119533,
+        "success_accuracy": 0.8653988838195801,
+        "total_loss": 1.6075192756123013
+      }
+    },
+    {
+      "epoch": 17,
+      "train": {
+        "bc_loss": 0.10401224603669511,
+        "field_effect_loss": 0.00939107199277108,
+        "field_potential_loss": 0.2336524592505561,
+        "field_preference_loss": 0.504797674715519,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.188088772404525,
+        "proposal_loss": 3.866093521764924e-06,
+        "rank_acc": 0.84073213653432,
+        "rank_loss": 0.504786646614472,
+        "regret_mae": 0.37325214263465667,
+        "success_accuracy": 0.8799120262265205,
+        "total_loss": 1.3481841501262453
+      },
+      "val": {
+        "bc_loss": 0.13761650315589374,
+        "field_effect_loss": 0.009955928712669346,
+        "field_potential_loss": 0.27949220190445584,
+        "field_preference_loss": 0.5152178886863921,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20229888955752054,
+        "proposal_loss": 3.732259478460441e-06,
+        "rank_acc": 0.8257863885826535,
+        "rank_loss": 0.5152055985397763,
+        "regret_mae": 0.3892310576306449,
+        "success_accuracy": 0.8631307383378347,
+        "total_loss": 1.5798539519309998
+      }
+    },
+    {
+      "epoch": 18,
+      "train": {
+        "bc_loss": 0.10218187033509214,
+        "field_effect_loss": 0.00937073394500961,
+        "field_potential_loss": 0.2300137784332037,
+        "field_preference_loss": 0.5043303022782007,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18780956955419648,
+        "proposal_loss": 3.812584537854491e-06,
+        "rank_acc": 0.8430086316333877,
+        "rank_loss": 0.5043176528480318,
+        "regret_mae": 0.36754630340470207,
+        "success_accuracy": 0.8807487818929884,
+        "total_loss": 1.3344386402103636
+      },
+      "val": {
+        "bc_loss": 0.13608192486895454,
+        "field_effect_loss": 0.009987579710367653,
+        "field_potential_loss": 0.28785979996124905,
+        "field_preference_loss": 0.5131366468138165,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19313485506508085,
+        "proposal_loss": 3.856972423616551e-06,
+        "rank_acc": 0.819874064789878,
+        "rank_loss": 0.5131295853190951,
+        "regret_mae": 0.35180893374813926,
+        "success_accuracy": 0.8696586555904813,
+        "total_loss": 1.5773619280921087
+      }
+    },
+    {
+      "epoch": 19,
+      "train": {
+        "bc_loss": 0.0961691427251531,
+        "field_effect_loss": 0.00936776307126921,
+        "field_potential_loss": 0.22697962075471878,
+        "field_preference_loss": 0.5031071503957113,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18863700268169245,
+        "proposal_loss": 3.896287915318277e-06,
+        "rank_acc": 0.8442784398794174,
+        "rank_loss": 0.5030944099028906,
+        "regret_mae": 0.3641950401167075,
+        "success_accuracy": 0.8793381916152107,
+        "total_loss": 1.299914435380035
+      },
+      "val": {
+        "bc_loss": 0.1342487372457981,
+        "field_effect_loss": 0.009969068587654166,
+        "field_potential_loss": 0.28760334187083775,
+        "field_preference_loss": 0.5093723485867182,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19713042262527677,
+        "proposal_loss": 3.6680396533483872e-06,
+        "rank_acc": 0.8306030399269528,
+        "rank_loss": 0.5093635337220298,
+        "regret_mae": 0.36566174195872414,
+        "success_accuracy": 0.8657314048873054,
+        "total_loss": 1.563755167855157
+      }
+    },
+    {
+      "epoch": 20,
+      "train": {
+        "bc_loss": 0.09341253914559881,
+        "field_effect_loss": 0.009322793753300276,
+        "field_potential_loss": 0.2226135316822264,
+        "field_preference_loss": 0.5010028829177221,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18691841906143558,
+        "proposal_loss": 3.843114623375894e-06,
+        "rank_acc": 0.8477198224928644,
+        "rank_loss": 0.5009899031784799,
+        "regret_mae": 0.36142161136700046,
+        "success_accuracy": 0.8809511893325381,
+        "total_loss": 1.2787491083145142
+      },
+      "val": {
+        "bc_loss": 0.13583754913674462,
+        "field_effect_loss": 0.010007145504156748,
+        "field_potential_loss": 0.2752663451764319,
+        "field_preference_loss": 0.5190565254953172,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20340084532896677,
+        "proposal_loss": 3.5167080088487837e-06,
+        "rank_acc": 0.8321969972716438,
+        "rank_loss": 0.5190456791056527,
+        "regret_mae": 0.36729014416535694,
+        "success_accuracy": 0.8631447388066186,
+        "total_loss": 1.571842994954851
+      }
+    },
+    {
+      "epoch": 21,
+      "train": {
+        "bc_loss": 0.08878885701091753,
+        "field_effect_loss": 0.009298105536597885,
+        "field_potential_loss": 0.2234488156520658,
+        "field_preference_loss": 0.500933326780796,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18844134277767605,
+        "proposal_loss": 3.6430501084952184e-06,
+        "rank_acc": 0.8479213656650649,
+        "rank_loss": 0.5009222018222014,
+        "regret_mae": 0.3631010159022278,
+        "success_accuracy": 0.8792463781105148,
+        "total_loss": 1.2564710941579607
+      },
+      "val": {
+        "bc_loss": 0.13547604613833958,
+        "field_effect_loss": 0.010036921749512354,
+        "field_potential_loss": 0.27759941501749885,
+        "field_preference_loss": 0.52272962530454,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19917072604099909,
+        "proposal_loss": 3.2815645580639184e-06,
+        "rank_acc": 0.823263535896937,
+        "rank_loss": 0.5227189924981859,
+        "regret_mae": 0.38013899160756004,
+        "success_accuracy": 0.871842794948154,
+        "total_loss": 1.5753508011500041
+      }
+    },
+    {
+      "epoch": 22,
+      "train": {
+        "bc_loss": 0.09232095209881663,
+        "field_effect_loss": 0.009306432084283896,
+        "field_potential_loss": 0.22280274435049957,
+        "field_preference_loss": 0.5005648670097192,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18803183382583988,
+        "proposal_loss": 3.515272500938712e-06,
+        "rank_acc": 0.8478854671120644,
+        "rank_loss": 0.5005538943741057,
+        "regret_mae": 0.36690422520041466,
+        "success_accuracy": 0.8809511893325381,
+        "total_loss": 1.2730992320511076
+      },
+      "val": {
+        "bc_loss": 0.14252920572956404,
+        "field_effect_loss": 0.009982850247373184,
+        "field_potential_loss": 0.2893659124771754,
+        "field_preference_loss": 0.5085958689451218,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1977451311217414,
+        "proposal_loss": 3.7967404422387417e-06,
+        "rank_acc": 0.8323784205648634,
+        "rank_loss": 0.5085858470863767,
+        "regret_mae": 0.3829231692685021,
+        "success_accuracy": 0.8729243642754025,
+        "total_loss": 1.6051508718066745
+      }
+    },
+    {
+      "epoch": 23,
+      "train": {
+        "bc_loss": 0.08938077159432901,
+        "field_effect_loss": 0.009281226000489874,
+        "field_potential_loss": 0.22155151805943912,
+        "field_preference_loss": 0.5004152179592185,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18713747647901377,
+        "proposal_loss": 3.651892279145209e-06,
+        "rank_acc": 0.8489571122659577,
+        "rank_loss": 0.5004063112040361,
+        "regret_mae": 0.3629038578106297,
+        "success_accuracy": 0.881247496439351,
+        "total_loss": 1.2566893680228128
+      },
+      "val": {
+        "bc_loss": 0.1322652217414644,
+        "field_effect_loss": 0.010026000388380554,
+        "field_potential_loss": 0.2885795517100228,
+        "field_preference_loss": 0.5062332368559308,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1950002759695053,
+        "proposal_loss": 3.3075696895442284e-06,
+        "rank_acc": 0.8296620117293464,
+        "rank_loss": 0.5062223523855209,
+        "regret_mae": 0.3615676628218757,
+        "success_accuracy": 0.8715242776605818,
+        "total_loss": 1.5516093638208177
+      }
+    },
+    {
+      "epoch": 24,
+      "train": {
+        "bc_loss": 0.0851077043140928,
+        "field_effect_loss": 0.009251353156287223,
+        "field_potential_loss": 0.2172074963649114,
+        "field_preference_loss": 0.4987656854920917,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18849724841614565,
+        "proposal_loss": 3.4045889599888646e-06,
+        "rank_acc": 0.8507693633437157,
+        "rank_loss": 0.498753914816512,
+        "regret_mae": 0.3611187262253629,
+        "success_accuracy": 0.8827832871013217,
+        "total_loss": 1.2282878243260913
+      },
+      "val": {
+        "bc_loss": 0.1346430571542846,
+        "field_effect_loss": 0.009975153021514416,
+        "field_potential_loss": 0.29989150332079995,
+        "field_preference_loss": 0.5020635409487618,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1960180484586292,
+        "proposal_loss": 3.338488833530088e-06,
+        "rank_acc": 0.83252904481358,
+        "rank_loss": 0.5020535969071918,
+        "regret_mae": 0.4121265212694804,
+        "success_accuracy": 0.8724938366148207,
+        "total_loss": 1.5680675903956096
+      }
+    },
+    {
+      "epoch": 25,
+      "train": {
+        "bc_loss": 0.08336237616216143,
+        "field_effect_loss": 0.009249880159687664,
+        "field_potential_loss": 0.21621989748544163,
+        "field_preference_loss": 0.4970288384291861,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18691645997265974,
+        "proposal_loss": 3.374253225274515e-06,
+        "rank_acc": 0.8525645443134837,
+        "rank_loss": 0.49701664679580265,
+        "regret_mae": 0.35812142656909096,
+        "success_accuracy": 0.8816064041521814,
+        "total_loss": 1.217036282022794
+      },
+      "val": {
+        "bc_loss": 0.13486805910037625,
+        "field_effect_loss": 0.01008807671152883,
+        "field_potential_loss": 0.2857277211215761,
+        "field_preference_loss": 0.5126529849237866,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19209459837940004,
+        "proposal_loss": 3.1567663856751297e-06,
+        "rank_acc": 0.8262526591618856,
+        "rank_loss": 0.5126433173815409,
+        "regret_mae": 0.3921952082051171,
+        "success_accuracy": 0.8795502914322747,
+        "total_loss": 1.5664001173443265
+      }
+    },
+    {
+      "epoch": 26,
+      "train": {
+        "bc_loss": 0.08421546572612391,
+        "field_effect_loss": 0.009248772668393536,
+        "field_potential_loss": 0.21602359186444017,
+        "field_preference_loss": 0.497691563434071,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1873969245288107,
+        "proposal_loss": 3.353415950060581e-06,
+        "rank_acc": 0.8520787705977758,
+        "rank_loss": 0.4976780807806386,
+        "regret_mae": 0.359428940133916,
+        "success_accuracy": 0.882996127837234,
+        "total_loss": 1.2216603342029784
+      },
+      "val": {
+        "bc_loss": 0.1341037050717407,
+        "field_effect_loss": 0.010077057871967554,
+        "field_potential_loss": 0.3124141643444697,
+        "field_preference_loss": 0.49701457222302753,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19009532034397125,
+        "proposal_loss": 3.6647965341924446e-06,
+        "rank_acc": 0.8260617189937167,
+        "rank_loss": 0.49700263970428044,
+        "regret_mae": 0.4092044648196962,
+        "success_accuracy": 0.8749929997656081,
+        "total_loss": 1.5734262863794963
+      }
+    },
+    {
+      "epoch": 27,
+      "train": {
+        "bc_loss": 0.08230337076303032,
+        "field_effect_loss": 0.009236639708332304,
+        "field_potential_loss": 0.21743292692634794,
+        "field_preference_loss": 0.4971166638036569,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18682393804192543,
+        "proposal_loss": 3.2966939740062016e-06,
+        "rank_acc": 0.8533231301440133,
+        "rank_loss": 0.4971059577332603,
+        "regret_mae": 0.36136048369937473,
+        "success_accuracy": 0.8823263057404094,
+        "total_loss": 1.212829314172268
+      },
+      "val": {
+        "bc_loss": 0.13366680488818222,
+        "field_effect_loss": 0.009977248238606585,
+        "field_potential_loss": 0.29063844432433444,
+        "field_preference_loss": 0.5076679786046346,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19655685043997234,
+        "proposal_loss": 3.300812522866181e-06,
+        "rank_acc": 0.8246373467975192,
+        "rank_loss": 0.5076571620172925,
+        "regret_mae": 0.3641776310073005,
+        "success_accuracy": 0.8686960935592651,
+        "total_loss": 1.5624717937575445
+      }
+    },
+    {
+      "epoch": 28,
+      "train": {
+        "bc_loss": 0.07730846303618616,
+        "field_effect_loss": 0.00920152369265755,
+        "field_potential_loss": 0.2122342342303859,
+        "field_preference_loss": 0.494945968190829,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18644743888742393,
+        "proposal_loss": 3.2203913886203536e-06,
+        "rank_acc": 0.8552364168895615,
+        "rank_loss": 0.49493074582682717,
+        "regret_mae": 0.3602006605101956,
+        "success_accuracy": 0.8817149110966258,
+        "total_loss": 1.1799213389555614
+      },
+      "val": {
+        "bc_loss": 0.13534324657585886,
+        "field_effect_loss": 0.009898109102828635,
+        "field_potential_loss": 0.2735137566924095,
+        "field_preference_loss": 0.5134736630651686,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.2001989268594318,
+        "proposal_loss": 2.66414412585113e-06,
+        "rank_acc": 0.8308547801441617,
+        "rank_loss": 0.5134652439090941,
+        "regret_mae": 0.3477882891893387,
+        "success_accuracy": 0.8684755795531802,
+        "total_loss": 1.5583282113075256
+      }
+    },
+    {
+      "epoch": 29,
+      "train": {
+        "bc_loss": 0.07689455071360701,
+        "field_effect_loss": 0.009122511831163947,
+        "field_potential_loss": 0.20597484728528392,
+        "field_preference_loss": 0.49515268289380604,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1874129699750079,
+        "proposal_loss": 3.1206082585160602e-06,
+        "rank_acc": 0.8557494555910429,
+        "rank_loss": 0.4951398844520251,
+        "regret_mae": 0.3543991752796703,
+        "success_accuracy": 0.8838913101289008,
+        "total_loss": 1.1713674714167912
+      },
+      "val": {
+        "bc_loss": 0.13915960159566668,
+        "field_effect_loss": 0.010020748298201296,
+        "field_potential_loss": 0.2832588048444854,
+        "field_preference_loss": 0.5201258957386017,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20523528340790007,
+        "proposal_loss": 2.973997526270169e-06,
+        "rank_acc": 0.8298549254735311,
+        "rank_loss": 0.5201137397024367,
+        "regret_mae": 0.41773803863260484,
+        "success_accuracy": 0.8723818295531802,
+        "total_loss": 1.5952921178605821
+      }
+    },
+    {
+      "epoch": 30,
+      "train": {
+        "bc_loss": 0.0738204204891291,
+        "field_effect_loss": 0.009115566583609002,
+        "field_potential_loss": 0.20450546871870756,
+        "field_preference_loss": 0.4932474129729801,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18662962627907595,
+        "proposal_loss": 3.225285764882655e-06,
+        "rank_acc": 0.8578701880243089,
+        "rank_loss": 0.49323253540529144,
+        "regret_mae": 0.35379838902089333,
+        "success_accuracy": 0.8848177923096551,
+        "total_loss": 1.151983039246665
+      },
+      "val": {
+        "bc_loss": 0.13380646684931385,
+        "field_effect_loss": 0.009914199718170695,
+        "field_potential_loss": 0.2761235965622796,
+        "field_preference_loss": 0.5150359719991684,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19994131889608172,
+        "proposal_loss": 3.0268790851550875e-06,
+        "rank_acc": 0.8282433085971408,
+        "rank_loss": 0.5150183853175905,
+        "regret_mae": 0.35405541128582424,
+        "success_accuracy": 0.8672855032814873,
+        "total_loss": 1.5567204223738775
+      }
+    },
+    {
+      "epoch": 31,
+      "train": {
+        "bc_loss": 0.07344370415537721,
+        "field_effect_loss": 0.009111036753488911,
+        "field_potential_loss": 0.20299633944200146,
+        "field_preference_loss": 0.492623714523183,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18506384847892654,
+        "proposal_loss": 2.9502330960440304e-06,
+        "rank_acc": 0.8598257559869025,
+        "rank_loss": 0.49261073023080826,
+        "regret_mae": 0.34877791214320397,
+        "success_accuracy": 0.8839434766107135,
+        "total_loss": 1.1477310177352693
+      },
+      "val": {
+        "bc_loss": 0.13509702537622717,
+        "field_effect_loss": 0.009968374959296651,
+        "field_potential_loss": 0.28459468483924866,
+        "field_preference_loss": 0.5111524098449283,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19811229242218864,
+        "proposal_loss": 2.9881120250744666e-06,
+        "rank_acc": 0.8295722603797913,
+        "rank_loss": 0.5111412637763553,
+        "regret_mae": 0.34907597965664333,
+        "success_accuracy": 0.8667429685592651,
+        "total_loss": 1.5660970012346904
+      }
+    },
+    {
+      "epoch": 32,
+      "train": {
+        "bc_loss": 0.07288210337153739,
+        "field_effect_loss": 0.009103896083413728,
+        "field_potential_loss": 0.20107182508541477,
+        "field_preference_loss": 0.4921406776540809,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18553692371480995,
+        "proposal_loss": 3.0461141086056564e-06,
+        "rank_acc": 0.8587863047917684,
+        "rank_loss": 0.49212736305263305,
+        "regret_mae": 0.3517974784804715,
+        "success_accuracy": 0.882943960527579,
+        "total_loss": 1.1423794354001682
+      },
+      "val": {
+        "bc_loss": 0.1359775825921032,
+        "field_effect_loss": 0.009947556847085556,
+        "field_potential_loss": 0.29057716329892475,
+        "field_preference_loss": 0.5089091393682692,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1966662241352929,
+        "proposal_loss": 3.178296727431492e-06,
+        "rank_acc": 0.830020374721951,
+        "rank_loss": 0.508900115887324,
+        "regret_mae": 0.3811758773194419,
+        "success_accuracy": 0.8733653955989413,
+        "total_loss": 1.572794086403317
+      }
+    },
+    {
+      "epoch": 33,
+      "train": {
+        "bc_loss": 0.07335518585104081,
+        "field_effect_loss": 0.00910466166290765,
+        "field_potential_loss": 0.2015460758573479,
+        "field_preference_loss": 0.492039542645216,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18661837673021686,
+        "proposal_loss": 3.0779953495615095e-06,
+        "rank_acc": 0.8593874449531237,
+        "rank_loss": 0.4920241903099749,
+        "regret_mae": 0.35099731882413227,
+        "success_accuracy": 0.8829898676938481,
+        "total_loss": 1.1452019669943385
+      },
+      "val": {
+        "bc_loss": 0.13693759300642544,
+        "field_effect_loss": 0.009944807272404432,
+        "field_potential_loss": 0.28522006008360123,
+        "field_preference_loss": 0.5129994799693426,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20066123041841719,
+        "proposal_loss": 2.70546226804274e-06,
+        "rank_acc": 0.8250638975037469,
+        "rank_loss": 0.5129875275823805,
+        "regret_mae": 0.38694268299473655,
+        "success_accuracy": 0.8683705727259318,
+        "total_loss": 1.5781877636909485
+      }
+    },
+    {
+      "epoch": 34,
+      "train": {
+        "bc_loss": 0.06938031245954335,
+        "field_effect_loss": 0.009131903087513315,
+        "field_potential_loss": 0.20100054372515944,
+        "field_preference_loss": 0.4922616754968961,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18540162013636696,
+        "proposal_loss": 2.9310312249385283e-06,
+        "rank_acc": 0.8580897409054968,
+        "rank_loss": 0.4922475351227654,
+        "regret_mae": 0.3494173288345337,
+        "success_accuracy": 0.8843149046103159,
+        "total_loss": 1.1251345798373222
+      },
+      "val": {
+        "bc_loss": 0.13860523245400852,
+        "field_effect_loss": 0.01002420966203014,
+        "field_potential_loss": 0.2973766442802217,
+        "field_preference_loss": 0.5044528775744967,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19975476380851534,
+        "proposal_loss": 2.9795605390366595e-06,
+        "rank_acc": 0.8332956035931905,
+        "rank_loss": 0.504441499710083,
+        "regret_mae": 0.427681823571523,
+        "success_accuracy": 0.8721823195616404,
+        "total_loss": 1.5891507400406732
+      }
+    },
+    {
+      "epoch": 35,
+      "train": {
+        "bc_loss": 0.07112570174245371,
+        "field_effect_loss": 0.009073995946285626,
+        "field_potential_loss": 0.19736613374617365,
+        "field_preference_loss": 0.48966583650973106,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1854082120375501,
+        "proposal_loss": 2.977692702188506e-06,
+        "rank_acc": 0.8630185135536723,
+        "rank_loss": 0.4896468313203918,
+        "regret_mae": 0.3559825089242723,
+        "success_accuracy": 0.8854124935136901,
+        "total_loss": 1.1268395707011223
+      },
+      "val": {
+        "bc_loss": 0.14179483180244765,
+        "field_effect_loss": 0.009873121459450986,
+        "field_potential_loss": 0.276170755426089,
+        "field_preference_loss": 0.5130939682324728,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1964685105615192,
+        "proposal_loss": 2.707515250222059e-06,
+        "rank_acc": 0.8302382893032498,
+        "rank_loss": 0.5130851053529315,
+        "regret_mae": 0.3540174629953172,
+        "success_accuracy": 0.8734739025433859,
+        "total_loss": 1.5926916864183214
+      }
+    },
+    {
+      "epoch": 36,
+      "train": {
+        "bc_loss": 0.07042640887407793,
+        "field_effect_loss": 0.009035933510555575,
+        "field_potential_loss": 0.19240674148831102,
+        "field_preference_loss": 0.4900285941031244,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1850507493234343,
+        "proposal_loss": 2.9707339876949845e-06,
+        "rank_acc": 0.8617810234427452,
+        "rank_loss": 0.4900139470895131,
+        "regret_mae": 0.34720859883560073,
+        "success_accuracy": 0.885504307018386,
+        "total_loss": 1.1183024073640506
+      },
+      "val": {
+        "bc_loss": 0.14694455017646155,
+        "field_effect_loss": 0.010058949080606302,
+        "field_potential_loss": 0.30040832940075135,
+        "field_preference_loss": 0.5029920091231664,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1902554945813285,
+        "proposal_loss": 3.254443148155891e-06,
+        "rank_acc": 0.8302126791742113,
+        "rank_loss": 0.5029788431194093,
+        "regret_mae": 0.3750624822245704,
+        "success_accuracy": 0.8697846664322747,
+        "total_loss": 1.6319977574878268
+      }
+    },
+    {
+      "epoch": 37,
+      "train": {
+        "bc_loss": 0.06777406408865419,
+        "field_effect_loss": 0.00905410039316242,
+        "field_potential_loss": 0.19368277138306034,
+        "field_preference_loss": 0.4881540880435043,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18435489986505774,
+        "proposal_loss": 2.887051814405418e-06,
+        "rank_acc": 0.8625231451458402,
+        "rank_loss": 0.48814112154973877,
+        "regret_mae": 0.3456342276185751,
+        "success_accuracy": 0.8829481345083978,
+        "total_loss": 1.1044291920132108
+      },
+      "val": {
+        "bc_loss": 0.13540590099162525,
+        "field_effect_loss": 0.009987057600584295,
+        "field_potential_loss": 0.3004314742154545,
+        "field_preference_loss": 0.5066264337963529,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1956225393546952,
+        "proposal_loss": 2.4653020950103787e-06,
+        "rank_acc": 0.8255961404906379,
+        "rank_loss": 0.5066133787234625,
+        "regret_mae": 0.40548118121094173,
+        "success_accuracy": 0.868150058719847,
+        "total_loss": 1.576615207725101
+      }
+    },
+    {
+      "epoch": 38,
+      "train": {
+        "bc_loss": 0.06584255014442736,
+        "field_effect_loss": 0.00901409016094274,
+        "field_potential_loss": 0.18798615638580587,
+        "field_preference_loss": 0.4883936548398601,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18413924984633923,
+        "proposal_loss": 2.77262412851087e-06,
+        "rank_acc": 0.8633801092704138,
+        "rank_loss": 0.48837929053439033,
+        "regret_mae": 0.3427066434588697,
+        "success_accuracy": 0.884705112212234,
+        "total_loss": 1.0889378885428111
+      },
+      "val": {
+        "bc_loss": 0.13667403265006012,
+        "field_effect_loss": 0.009956308485319218,
+        "field_potential_loss": 0.2826397054725223,
+        "field_preference_loss": 0.5156110160880618,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19597034653027853,
+        "proposal_loss": 2.7768909250072384e-06,
+        "rank_acc": 0.8288223346074423,
+        "rank_loss": 0.5156038221385744,
+        "regret_mae": 0.34998622205522323,
+        "success_accuracy": 0.8645728296703763,
+        "total_loss": 1.5773590869373746
+      }
+    },
+    {
+      "epoch": 39,
+      "train": {
+        "bc_loss": 0.0637634849279291,
+        "field_effect_loss": 0.00901427192406522,
+        "field_potential_loss": 0.19114136126720244,
+        "field_preference_loss": 0.4876866688330968,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1838830326580339,
+        "proposal_loss": 2.8208763656796996e-06,
+        "rank_acc": 0.8654589859975709,
+        "rank_loss": 0.48767346060938305,
+        "regret_mae": 0.3492264387508233,
+        "success_accuracy": 0.8874052655365732,
+        "total_loss": 1.080943975183699
+      },
+      "val": {
+        "bc_loss": 0.14109190098113483,
+        "field_effect_loss": 0.00997752541055282,
+        "field_potential_loss": 0.2878431859943602,
+        "field_preference_loss": 0.5120130644904243,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1930425324373775,
+        "proposal_loss": 2.761687558934985e-06,
+        "rank_acc": 0.8289885951413049,
+        "rank_loss": 0.5120042910178503,
+        "regret_mae": 0.3591463267803192,
+        "success_accuracy": 0.8705372114976248,
+        "total_loss": 1.6005899045202467
+      }
+    },
+    {
+      "epoch": 40,
+      "train": {
+        "bc_loss": 0.06218210944078035,
+        "field_effect_loss": 0.009005796966246433,
+        "field_potential_loss": 0.18933595220247904,
+        "field_preference_loss": 0.4858348634507921,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18345809769299296,
+        "proposal_loss": 2.817829119471248e-06,
+        "rank_acc": 0.8657766133546829,
+        "rank_loss": 0.4858182333409786,
+        "regret_mae": 0.3526029785474141,
+        "success_accuracy": 0.8853791074620353,
+        "total_loss": 1.0692620757553313
+      },
+      "val": {
+        "bc_loss": 0.13362486329343584,
+        "field_effect_loss": 0.009969199748916758,
+        "field_potential_loss": 0.29624341511064106,
+        "field_preference_loss": 0.5119189835257001,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19776060597764122,
+        "proposal_loss": 2.830297590763722e-06,
+        "rank_acc": 0.8278980950514475,
+        "rank_loss": 0.511908181839519,
+        "regret_mae": 0.3915792968538072,
+        "success_accuracy": 0.8712967601087358,
+        "total_loss": 1.5694955521159701
+      }
+    },
+    {
+      "epoch": 41,
+      "train": {
+        "bc_loss": 0.06323594325739476,
+        "field_effect_loss": 0.008974810680633204,
+        "field_potential_loss": 0.18394408519897196,
+        "field_preference_loss": 0.4859275155597263,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1840947214514017,
+        "proposal_loss": 2.7925252968291006e-06,
+        "rank_acc": 0.8670085238085853,
+        "rank_loss": 0.48591120251350933,
+        "regret_mae": 0.3418093944589297,
+        "success_accuracy": 0.8861908225549592,
+        "total_loss": 1.068665844698747
+      },
+      "val": {
+        "bc_loss": 0.14023515292339855,
+        "field_effect_loss": 0.010029718279838562,
+        "field_potential_loss": 0.31382252938217586,
+        "field_preference_loss": 0.5017670657899644,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19935349043872622,
+        "proposal_loss": 3.2837739101725874e-06,
+        "rank_acc": 0.8262166844473945,
+        "rank_loss": 0.5017552889055676,
+        "regret_mae": 0.4291742725504769,
+        "success_accuracy": 0.8672890067100525,
+        "total_loss": 1.6100677649180095
+      }
+    },
+    {
+      "epoch": 42,
+      "train": {
+        "bc_loss": 0.06514720836033423,
+        "field_effect_loss": 0.008981518408594033,
+        "field_potential_loss": 0.18410294089052412,
+        "field_preference_loss": 0.4852833069033093,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1828850642260578,
+        "proposal_loss": 2.9065154952453063e-06,
+        "rank_acc": 0.8663367239965333,
+        "rank_loss": 0.4852669719192717,
+        "regret_mae": 0.3408734868797991,
+        "success_accuracy": 0.8845193982124329,
+        "total_loss": 1.0777163464162085
+      },
+      "val": {
+        "bc_loss": 0.14034779287046856,
+        "field_effect_loss": 0.009949027560651302,
+        "field_potential_loss": 0.29164400200049084,
+        "field_preference_loss": 0.5110442158248689,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19612882203525966,
+        "proposal_loss": 3.0120125605511325e-06,
+        "rank_acc": 0.8288029332955679,
+        "rank_loss": 0.5110359125667148,
+        "regret_mae": 0.36009517974323696,
+        "success_accuracy": 0.8689201109939151,
+        "total_loss": 1.5986986358960469
+      }
+    },
+    {
+      "epoch": 43,
+      "train": {
+        "bc_loss": 0.0616382923649831,
+        "field_effect_loss": 0.008932564045406051,
+        "field_potential_loss": 0.175693499441776,
+        "field_preference_loss": 0.4832448682023419,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1835053830097119,
+        "proposal_loss": 2.8897722820096938e-06,
+        "rank_acc": 0.8701142312751876,
+        "rank_loss": 0.4832271258864138,
+        "regret_mae": 0.343448215474685,
+        "success_accuracy": 0.88739691923062,
+        "total_loss": 1.0490744329161115
+      },
+      "val": {
+        "bc_loss": 0.13605627438260448,
+        "field_effect_loss": 0.01000926886788673,
+        "field_potential_loss": 0.2910189810726378,
+        "field_preference_loss": 0.5192269169621997,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19811676939328512,
+        "proposal_loss": 2.6895437612337346e-06,
+        "rank_acc": 0.8275181088182662,
+        "rank_loss": 0.519215957985984,
+        "regret_mae": 0.3527845839659373,
+        "success_accuracy": 0.8674920168187883,
+        "total_loss": 1.5856226483980815
+      }
+    },
+    {
+      "epoch": 44,
+      "train": {
+        "bc_loss": 0.06127763787905375,
+        "field_effect_loss": 0.00892588981272032,
+        "field_potential_loss": 0.17616962010247839,
+        "field_preference_loss": 0.48226701385445064,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18312197261386448,
+        "proposal_loss": 2.7046772730601433e-06,
+        "rank_acc": 0.8714442741539743,
+        "rank_loss": 0.4822527861429585,
+        "regret_mae": 0.3408530068894227,
+        "success_accuracy": 0.8885216348701053,
+        "total_loss": 1.0467579662799835
+      },
+      "val": {
+        "bc_loss": 0.1360557104150454,
+        "field_effect_loss": 0.010039205714646313,
+        "field_potential_loss": 0.2942740296324094,
+        "field_preference_loss": 0.5201205776797401,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19961999108393988,
+        "proposal_loss": 3.440809362271264e-06,
+        "rank_acc": 0.8234045141273074,
+        "rank_loss": 0.5201147132449679,
+        "regret_mae": 0.3741459929280811,
+        "success_accuracy": 0.8657664060592651,
+        "total_loss": 1.589431921641032
+      }
+    },
+    {
+      "epoch": 45,
+      "train": {
+        "bc_loss": 0.059481978701013655,
+        "field_effect_loss": 0.008905288653396484,
+        "field_potential_loss": 0.17420968496137196,
+        "field_preference_loss": 0.48179668312271434,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1827742097278436,
+        "proposal_loss": 2.784503859882332e-06,
+        "rank_acc": 0.8718954490290748,
+        "rank_loss": 0.481782844911019,
+        "regret_mae": 0.34136109178264934,
+        "success_accuracy": 0.886361929277579,
+        "total_loss": 1.034982951151
+      },
+      "val": {
+        "bc_loss": 0.1346837313224872,
+        "field_effect_loss": 0.010024741809401248,
+        "field_potential_loss": 0.2994039828578631,
+        "field_preference_loss": 0.5102392567528619,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19765183991856045,
+        "proposal_loss": 2.752743411191558e-06,
+        "rank_acc": 0.8287612431579165,
+        "rank_loss": 0.5102331886688868,
+        "regret_mae": 0.38694362176789177,
+        "success_accuracy": 0.8706422183248732,
+        "total_loss": 1.5766458776262071
+      }
+    },
+    {
+      "epoch": 46,
+      "train": {
+        "bc_loss": 0.057989504343519606,
+        "field_effect_loss": 0.008895731349993084,
+        "field_potential_loss": 0.17016280659784874,
+        "field_preference_loss": 0.4800613671541214,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1823639197068082,
+        "proposal_loss": 2.7561112132288044e-06,
+        "rank_acc": 0.8729064481125938,
+        "rank_loss": 0.48003895911905503,
+        "regret_mae": 0.3415820805562867,
+        "success_accuracy": 0.890658386879497,
+        "total_loss": 1.0213559890786807
+      },
+      "val": {
+        "bc_loss": 0.13971792492601606,
+        "field_effect_loss": 0.01000478035873837,
+        "field_potential_loss": 0.29788532935910755,
+        "field_preference_loss": 0.5198098437653648,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19898550543520185,
+        "proposal_loss": 2.677549964068425e-06,
+        "rank_acc": 0.8216269612312317,
+        "rank_loss": 0.519804072048929,
+        "regret_mae": 0.37061287297142875,
+        "success_accuracy": 0.8740164372656081,
+        "total_loss": 1.6121906969282362
+      }
+    },
+    {
+      "epoch": 47,
+      "train": {
+        "bc_loss": 0.057118677663513355,
+        "field_effect_loss": 0.00885501406284877,
+        "field_potential_loss": 0.1665032353873054,
+        "field_preference_loss": 0.47937826895051533,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18151584412488672,
+        "proposal_loss": 2.7964930811524228e-06,
+        "rank_acc": 0.8751616668370035,
+        "rank_loss": 0.4793611992564466,
+        "regret_mae": 0.34056081084741485,
+        "success_accuracy": 0.8921879173980819,
+        "total_loss": 1.0123616713616583
+      },
+      "val": {
+        "bc_loss": 0.1387146388490995,
+        "field_effect_loss": 0.010124866747193866,
+        "field_potential_loss": 0.3061202284362581,
+        "field_preference_loss": 0.5172444234291712,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20290558371278974,
+        "proposal_loss": 2.8749063226819594e-06,
+        "rank_acc": 0.8258393042617374,
+        "rank_loss": 0.5172349082099067,
+        "regret_mae": 0.3965615861945682,
+        "success_accuracy": 0.8651083641582065,
+        "total_loss": 1.613426148891449
+      }
+    },
+    {
+      "epoch": 48,
+      "train": {
+        "bc_loss": 0.05922183350452946,
+        "field_effect_loss": 0.008864264792969657,
+        "field_potential_loss": 0.16684888758593136,
+        "field_preference_loss": 0.4793485804564423,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18174966507487827,
+        "proposal_loss": 2.7044819274932606e-06,
+        "rank_acc": 0.8753615096211433,
+        "rank_loss": 0.47932569889558685,
+        "regret_mae": 0.3350806397696336,
+        "success_accuracy": 0.889554537832737,
+        "total_loss": 1.0230107423332002
+      },
+      "val": {
+        "bc_loss": 0.13975584589772755,
+        "field_effect_loss": 0.010137717498259412,
+        "field_potential_loss": 0.3026132881641388,
+        "field_preference_loss": 0.5147485832373301,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19820464899142584,
+        "proposal_loss": 2.405162743141571e-06,
+        "rank_acc": 0.8238838679260678,
+        "rank_loss": 0.5147323078579373,
+        "regret_mae": 0.3682815068297916,
+        "success_accuracy": 0.8664139476087358,
+        "total_loss": 1.6123907897207472
+      }
+    },
+    {
+      "epoch": 49,
+      "train": {
+        "bc_loss": 0.05752872973163095,
+        "field_effect_loss": 0.00886419784769209,
+        "field_potential_loss": 0.1664471177177297,
+        "field_preference_loss": 0.4788004979491234,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18046389147639275,
+        "proposal_loss": 2.643351583249872e-06,
+        "rank_acc": 0.8749347221520212,
+        "rank_loss": 0.47878100184930694,
+        "regret_mae": 0.3374463795787758,
+        "success_accuracy": 0.8924278848701053,
+        "total_loss": 1.0135799621542294
+      },
+      "val": {
+        "bc_loss": 0.13783947295612758,
+        "field_effect_loss": 0.010076353067739142,
+        "field_potential_loss": 0.3024477909008662,
+        "field_preference_loss": 0.5125124239259295,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20247557924853432,
+        "proposal_loss": 2.5332821375438166e-06,
+        "rank_acc": 0.8251384231779311,
+        "rank_loss": 0.5124957710504532,
+        "regret_mae": 0.39886628091335297,
+        "success_accuracy": 0.8684825830989413,
+        "total_loss": 1.5988854964574177
+      }
+    },
+    {
+      "epoch": 50,
+      "train": {
+        "bc_loss": 0.05496217144860162,
+        "field_effect_loss": 0.008809831258582158,
+        "field_potential_loss": 0.15968970923374096,
+        "field_preference_loss": 0.47663103747699,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18219435732397768,
+        "proposal_loss": 2.634354959556933e-06,
+        "rank_acc": 0.8774759843945503,
+        "rank_loss": 0.47661108896136284,
+        "regret_mae": 0.33785633949769867,
+        "success_accuracy": 0.8918895233008597,
+        "total_loss": 0.9912039124303393
+      },
+      "val": {
+        "bc_loss": 0.13703543920483854,
+        "field_effect_loss": 0.010142495286547475,
+        "field_potential_loss": 0.3259502649307251,
+        "field_preference_loss": 0.5082890954282548,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19784927782085207,
+        "proposal_loss": 2.5641976561827405e-06,
+        "rank_acc": 0.8208600481351217,
+        "rank_loss": 0.5082813799381256,
+        "regret_mae": 0.4210789038075341,
+        "success_accuracy": 0.8697776628865136,
+        "total_loss": 1.6132541828685336
+      }
+    }
+  ]
+}

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_1/resolved_config.json

@@ -0,0 +1,69 @@
+{
+  "action_dim": 7,
+  "action_horizon": 16,
+  "backbone_feature_batch_size": 64,
+  "backbone_feature_cache": null,
+  "backbone_freeze": true,
+  "backbone_local_files_only": true,
+  "backbone_model": null,
+  "backbone_type": "native",
+  "batch_groups": 32,
+  "batch_size_groups": null,
+  "dataset_dir": "/scratch/knguy52/dovla/experiments/h16_merged_dataset",
+  "device": "cuda",
+  "effect_dim": 32,
+  "epochs": 50,
+  "hidden_dim": 256,
+  "lang_dim": 64,
+  "lattice_neighbors": 32,
+  "learning_rate": 0.001,
+  "losses": {
+    "bc": 5.0,
+    "bc_best_action": 1.0,
+    "causal_contrastive": 0.5,
+    "contrast": 0.5,
+    "effect": 1.0,
+    "field_anchor": 0.25,
+    "field_effect": 1.0,
+    "field_potential": 1.0,
+    "field_preference": 1.0,
+    "forward_effect_prediction": 1.0,
+    "lang_pair": 0.25,
+    "language_minimal_pair": 0.25,
+    "progress": 1.0,
+    "proposal": 2.0,
+    "rank": 1.0,
+    "regret": 0.5,
+    "regret_prediction": 0.5,
+    "same_state_pairwise_ranking": 1.0,
+    "success": 1.0
+  },
+  "lr": null,
+  "objective": "lattice_field",
+  "obs_dim": 96,
+  "observation_mode": "state",
+  "output_dir": "/scratch/knguy52/dovla/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_1",
+  "pair_count_per_group": 32,
+  "pair_scope": "same_state",
+  "policy_target_map": null,
+  "policy_target_map_coverage": {
+    "num_targets": 0,
+    "train_coverage": 0.0,
+    "train_covered": 0,
+    "train_groups": 2298,
+    "val_coverage": 0.0,
+    "val_covered": 0,
+    "val_groups": 575
+  },
+  "policy_target_types": [
+    "near_miss"
+  ],
+  "proposal_types": [
+    "no_op"
+  ],
+  "records_per_group": 16,
+  "seed": 1,
+  "val_fraction": 0.2,
+  "wandb": false,
+  "weight_decay": 0.0
+}

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_2/best.pt

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_2/best_policy.pt

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_2/latest.pt

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

scratch/experiments/dovla_h16_policy_ckpt_runs/near_miss_policy_bc5_typedprop_nooponly_p2/seed_2/metrics.json

@@ -0,0 +1,1582 @@
+{
+  "best": {
+    "bc_loss": 0.11740237764186329,
+    "field_effect_loss": 0.009628279315721657,
+    "field_potential_loss": 0.27409548146857154,
+    "field_preference_loss": 0.5002319051159753,
+    "lattice_edges": 3833.3333333333335,
+    "progress_mae": 0.19626346561643812,
+    "proposal_loss": 2.5106690928724775e-06,
+    "rank_acc": 0.8372291227181753,
+    "rank_loss": 0.5001979172229767,
+    "regret_mae": 0.38190865682231057,
+    "success_accuracy": 0.8718812995486789,
+    "total_loss": 1.4539916515350342
+  },
+  "best_policy": {
+    "bc_loss": 0.11538442307048374,
+    "field_effect_loss": 0.00973979427686168,
+    "field_potential_loss": 0.2751605494154824,
+    "field_preference_loss": 0.5092001971271303,
+    "lattice_edges": 3833.3333333333335,
+    "progress_mae": 0.198434103694227,
+    "proposal_loss": 3.109688893548284e-06,
+    "rank_acc": 0.833484646346834,
+    "rank_loss": 0.5091651446289487,
+    "regret_mae": 0.3825343433353636,
+    "success_accuracy": 0.8670964936415354,
+    "total_loss": 1.4539120992024739
+  },
+  "history": [
+    {
+      "epoch": 1,
+      "train": {
+        "bc_loss": 0.3173494657708539,
+        "field_effect_loss": 0.012260058752468063,
+        "field_potential_loss": 0.3728942457172606,
+        "field_preference_loss": 0.5573585339718394,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.20487802496386898,
+        "proposal_loss": 0.0009042316324767955,
+        "rank_acc": 0.7706687210334672,
+        "rank_loss": 0.5573620920379957,
+        "regret_mae": 0.6384340185258124,
+        "success_accuracy": 0.8563034195039008,
+        "total_loss": 2.639063321881824
+      },
+      "val": {
+        "bc_loss": 0.21573073085811403,
+        "field_effect_loss": 0.01096300713510977,
+        "field_potential_loss": 0.35152893596225315,
+        "field_preference_loss": 0.5539790656831529,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20879939115709728,
+        "proposal_loss": 0.00010837876147383617,
+        "rank_acc": 0.783986230691274,
+        "rank_loss": 0.5539556443691254,
+        "regret_mae": 0.5629676663213306,
+        "success_accuracy": 0.8611041108767191,
+        "total_loss": 2.0933103031582303
+      }
+    },
+    {
+      "epoch": 2,
+      "train": {
+        "bc_loss": 0.2107960283756256,
+        "field_effect_loss": 0.010735858115367591,
+        "field_potential_loss": 0.32964292189313305,
+        "field_preference_loss": 0.5387168468700515,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.19251438726981482,
+        "proposal_loss": 7.431919246098712e-05,
+        "rank_acc": 0.7969583223263422,
+        "rank_loss": 0.5387170356180933,
+        "regret_mae": 0.5452588295771016,
+        "success_accuracy": 0.8668974530365732,
+        "total_loss": 2.0285004129012427
+      },
+      "val": {
+        "bc_loss": 0.18780725946029028,
+        "field_effect_loss": 0.01080771168279979,
+        "field_potential_loss": 0.3216118746333652,
+        "field_preference_loss": 0.5454323523574405,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.22069531182448068,
+        "proposal_loss": 5.206997431944021e-05,
+        "rank_acc": 0.8007274137602912,
+        "rank_loss": 0.545406656132804,
+        "regret_mae": 0.5545442601044973,
+        "success_accuracy": 0.8574148747656081,
+        "total_loss": 1.9147024552027385
+      }
+    },
+    {
+      "epoch": 3,
+      "train": {
+        "bc_loss": 0.18742616350452104,
+        "field_effect_loss": 0.010457956154520312,
+        "field_potential_loss": 0.30892051632205647,
+        "field_preference_loss": 0.5300153771208392,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.19125411038597426,
+        "proposal_loss": 4.279871152195685e-05,
+        "rank_acc": 0.8059285432100296,
+        "rank_loss": 0.5300156838364072,
+        "regret_mae": 0.4826265660425027,
+        "success_accuracy": 0.8650215350919299,
+        "total_loss": 1.8785033408138487
+      },
+      "val": {
+        "bc_loss": 0.17164618397752443,
+        "field_effect_loss": 0.010418408498581912,
+        "field_potential_loss": 0.30404964751667446,
+        "field_preference_loss": 0.5244678424464332,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.2008535158303049,
+        "proposal_loss": 3.218912934244145e-05,
+        "rank_acc": 0.8053570720884535,
+        "rank_loss": 0.5244395915004942,
+        "regret_mae": 0.411051192217403,
+        "success_accuracy": 0.8579609096050262,
+        "total_loss": 1.7888358500268724
+      }
+    },
+    {
+      "epoch": 4,
+      "train": {
+        "bc_loss": 0.1739532167298926,
+        "field_effect_loss": 0.010258900622526804,
+        "field_potential_loss": 0.28762802709307933,
+        "field_preference_loss": 0.5243688291973538,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18905924322704473,
+        "proposal_loss": 3.0557374101489484e-05,
+        "rank_acc": 0.81168768223789,
+        "rank_loss": 0.5243671317067411,
+        "regret_mae": 0.4429139834311273,
+        "success_accuracy": 0.8672459274530411,
+        "total_loss": 1.7813684542973836
+      },
+      "val": {
+        "bc_loss": 0.1574412476685312,
+        "field_effect_loss": 0.010276101477858093,
+        "field_potential_loss": 0.28755974852376515,
+        "field_preference_loss": 0.5250948584742017,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20225257674853006,
+        "proposal_loss": 2.6142582505094146e-05,
+        "rank_acc": 0.8076592485109965,
+        "rank_loss": 0.5250626653432846,
+        "regret_mae": 0.42126401099893784,
+        "success_accuracy": 0.8649123542838626,
+        "total_loss": 1.700499501493242
+      }
+    },
+    {
+      "epoch": 5,
+      "train": {
+        "bc_loss": 0.1598427528515458,
+        "field_effect_loss": 0.010119614230158428,
+        "field_potential_loss": 0.27988324955933624,
+        "field_preference_loss": 0.5223573760853873,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.19014579181869826,
+        "proposal_loss": 2.220763831751861e-05,
+        "rank_acc": 0.8156778812408447,
+        "rank_loss": 0.5223552915785048,
+        "regret_mae": 0.42711034913857776,
+        "success_accuracy": 0.8693012992540995,
+        "total_loss": 1.6990843845738306
+      },
+      "val": {
+        "bc_loss": 0.14903792863090834,
+        "field_effect_loss": 0.01025437998274962,
+        "field_potential_loss": 0.2817705190844006,
+        "field_preference_loss": 0.5321706334749857,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1987346021665467,
+        "proposal_loss": 1.85911209579596e-05,
+        "rank_acc": 0.8126603199376,
+        "rank_loss": 0.5321383939849006,
+        "regret_mae": 0.48170561095078784,
+        "success_accuracy": 0.8647058407465616,
+        "total_loss": 1.6619042091899447
+      }
+    },
+    {
+      "epoch": 6,
+      "train": {
+        "bc_loss": 0.15238010769502985,
+        "field_effect_loss": 0.010016409253391126,
+        "field_potential_loss": 0.2744684952000777,
+        "field_preference_loss": 0.5201444170541234,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18971846107807425,
+        "proposal_loss": 1.7276948306567243e-05,
+        "rank_acc": 0.8172038280301623,
+        "rank_loss": 0.5201434103979005,
+        "regret_mae": 0.420024289853043,
+        "success_accuracy": 0.8696789865692457,
+        "total_loss": 1.653598467508952
+      },
+      "val": {
+        "bc_loss": 0.1446043530272113,
+        "field_effect_loss": 0.010134469096859297,
+        "field_potential_loss": 0.2818703196114964,
+        "field_preference_loss": 0.521719710694419,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1988751838604609,
+        "proposal_loss": 1.448093043008056e-05,
+        "rank_acc": 0.8173085451126099,
+        "rank_loss": 0.5216858983039856,
+        "regret_mae": 0.42713481684525806,
+        "success_accuracy": 0.8688431051042345,
+        "total_loss": 1.6259359452459548
+      }
+    },
+    {
+      "epoch": 7,
+      "train": {
+        "bc_loss": 0.14711939947058758,
+        "field_effect_loss": 0.009913377664310651,
+        "field_potential_loss": 0.2668979387316439,
+        "field_preference_loss": 0.5175117601950964,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1885704141524103,
+        "proposal_loss": 1.336063934963426e-05,
+        "rank_acc": 0.8201279739538828,
+        "rank_loss": 0.5175100937485695,
+        "regret_mae": 0.4005505782034662,
+        "success_accuracy": 0.8736290567451053,
+        "total_loss": 1.6153952694601483
+      },
+      "val": {
+        "bc_loss": 0.1485476911895805,
+        "field_effect_loss": 0.010050216586225562,
+        "field_potential_loss": 0.2771861031651497,
+        "field_preference_loss": 0.52001467678282,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20021594481335747,
+        "proposal_loss": 1.1985439313219911e-05,
+        "rank_acc": 0.8149468799432119,
+        "rank_loss": 0.519975913895501,
+        "regret_mae": 0.41706518994437325,
+        "success_accuracy": 0.8691616257031759,
+        "total_loss": 1.6382901337411668
+      }
+    },
+    {
+      "epoch": 8,
+      "train": {
+        "bc_loss": 0.14070771055089104,
+        "field_effect_loss": 0.009793190596004328,
+        "field_potential_loss": 0.2589014325704839,
+        "field_preference_loss": 0.5161434921125571,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18851213488313887,
+        "proposal_loss": 1.0060080196833749e-05,
+        "rank_acc": 0.8227507728669379,
+        "rank_loss": 0.5161437061097887,
+        "regret_mae": 0.3906004937986533,
+        "success_accuracy": 0.874536759323544,
+        "total_loss": 1.5723206599553425
+      },
+      "val": {
+        "bc_loss": 0.1418252376218637,
+        "field_effect_loss": 0.010024969698861241,
+        "field_potential_loss": 0.28779931200875175,
+        "field_preference_loss": 0.5121549185779359,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19227260102828345,
+        "proposal_loss": 9.636702290865489e-06,
+        "rank_acc": 0.8162082367473178,
+        "rank_loss": 0.5121151589685016,
+        "regret_mae": 0.39731647074222565,
+        "success_accuracy": 0.8820914526780447,
+        "total_loss": 1.6051721506648593
+      }
+    },
+    {
+      "epoch": 9,
+      "train": {
+        "bc_loss": 0.13737708381894562,
+        "field_effect_loss": 0.009750374159516973,
+        "field_potential_loss": 0.2576536503103044,
+        "field_preference_loss": 0.5147244454258018,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1898491537819306,
+        "proposal_loss": 9.16448199303785e-06,
+        "rank_acc": 0.8265425430403815,
+        "rank_loss": 0.5147217056817479,
+        "regret_mae": 0.3894164351125558,
+        "success_accuracy": 0.8766964649160703,
+        "total_loss": 1.5524456815587149
+      },
+      "val": {
+        "bc_loss": 0.14296618766254848,
+        "field_effect_loss": 0.00988098316722446,
+        "field_potential_loss": 0.2715096672375997,
+        "field_preference_loss": 0.5154482672611872,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19425899121496412,
+        "proposal_loss": 7.884352650459428e-06,
+        "rank_acc": 0.8252151177989112,
+        "rank_loss": 0.5154139548540115,
+        "regret_mae": 0.36880362696117824,
+        "success_accuracy": 0.8686225910981497,
+        "total_loss": 1.598752412531111
+      }
+    },
+    {
+      "epoch": 10,
+      "train": {
+        "bc_loss": 0.1324931506274475,
+        "field_effect_loss": 0.00967442627168364,
+        "field_potential_loss": 0.25208679648737115,
+        "field_preference_loss": 0.5123185387088193,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18730524741113186,
+        "proposal_loss": 7.629523483703653e-06,
+        "rank_acc": 0.8284097694688373,
+        "rank_loss": 0.5123143469293913,
+        "regret_mae": 0.38469482875532574,
+        "success_accuracy": 0.878305288652579,
+        "total_loss": 1.5193448331620958
+      },
+      "val": {
+        "bc_loss": 0.13597388399971855,
+        "field_effect_loss": 0.009897336079221632,
+        "field_potential_loss": 0.2819862614075343,
+        "field_preference_loss": 0.5137341519196829,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1977966527144114,
+        "proposal_loss": 7.14106113264279e-06,
+        "rank_acc": 0.8252179887559679,
+        "rank_loss": 0.513696637418535,
+        "regret_mae": 0.36611641115612453,
+        "success_accuracy": 0.8665469586849213,
+        "total_loss": 1.572011583381229
+      }
+    },
+    {
+      "epoch": 11,
+      "train": {
+        "bc_loss": 0.125258335740202,
+        "field_effect_loss": 0.009628132263767637,
+        "field_potential_loss": 0.2509075589478016,
+        "field_preference_loss": 0.5118478644225333,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.19043628767960602,
+        "proposal_loss": 6.3121584806443325e-06,
+        "rank_acc": 0.8318515618642172,
+        "rank_loss": 0.5118442508909438,
+        "regret_mae": 0.3904183833963341,
+        "success_accuracy": 0.8766797714763217,
+        "total_loss": 1.4815027018388112
+      },
+      "val": {
+        "bc_loss": 0.13785556786590153,
+        "field_effect_loss": 0.010054809403502278,
+        "field_potential_loss": 0.3089815576871236,
+        "field_preference_loss": 0.4980891876750522,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.18824820468823114,
+        "proposal_loss": 5.89844696858361e-06,
+        "rank_acc": 0.8270786338382297,
+        "rank_loss": 0.4980488005611632,
+        "regret_mae": 0.39049576057328117,
+        "success_accuracy": 0.8696026537153456,
+        "total_loss": 1.5909695492850409
+      }
+    },
+    {
+      "epoch": 12,
+      "train": {
+        "bc_loss": 0.12024382212095791,
+        "field_effect_loss": 0.009610261962128183,
+        "field_potential_loss": 0.24788866916464436,
+        "field_preference_loss": 0.5106851504080825,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18948376240829626,
+        "proposal_loss": 5.611595794461108e-06,
+        "rank_acc": 0.8330933252970377,
+        "rank_loss": 0.5106817425953017,
+        "regret_mae": 0.38483728882339263,
+        "success_accuracy": 0.8772473459442457,
+        "total_loss": 1.4515546709299088
+      },
+      "val": {
+        "bc_loss": 0.12894783541560173,
+        "field_effect_loss": 0.009772037414626943,
+        "field_potential_loss": 0.2834008245004548,
+        "field_preference_loss": 0.5034412658876843,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19281457364559174,
+        "proposal_loss": 5.203668630808049e-06,
+        "rank_acc": 0.8289878202809228,
+        "rank_loss": 0.5034036967489455,
+        "regret_mae": 0.4267410222027037,
+        "success_accuracy": 0.8737294177214304,
+        "total_loss": 1.5256295270389981
+      }
+    },
+    {
+      "epoch": 13,
+      "train": {
+        "bc_loss": 0.1203284340703653,
+        "field_effect_loss": 0.00953240933853926,
+        "field_potential_loss": 0.24308784844146836,
+        "field_preference_loss": 0.5093168562485112,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1873162957943148,
+        "proposal_loss": 5.054416331202245e-06,
+        "rank_acc": 0.8342370664079984,
+        "rank_loss": 0.5093152018056976,
+        "regret_mae": 0.3739333483907912,
+        "success_accuracy": 0.8804483008053567,
+        "total_loss": 1.4447136438555188
+      },
+      "val": {
+        "bc_loss": 0.1360906246635649,
+        "field_effect_loss": 0.009728612662810419,
+        "field_potential_loss": 0.26110465824604034,
+        "field_preference_loss": 0.5182190719578001,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19583768314785427,
+        "proposal_loss": 4.645822779518009e-06,
+        "rank_acc": 0.8293437361717224,
+        "rank_loss": 0.518183516131507,
+        "regret_mae": 0.3719827930132548,
+        "success_accuracy": 0.869921174314287,
+        "total_loss": 1.5565179851320055
+      }
+    },
+    {
+      "epoch": 14,
+      "train": {
+        "bc_loss": 0.11534101505660349,
+        "field_effect_loss": 0.00952825772239723,
+        "field_potential_loss": 0.24242615637679896,
+        "field_preference_loss": 0.5082112480368879,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18739925055868095,
+        "proposal_loss": 4.624387369429112e-06,
+        "rank_acc": 0.8376933162411054,
+        "rank_loss": 0.5082072532839246,
+        "regret_mae": 0.3747139010164473,
+        "success_accuracy": 0.8786537630690469,
+        "total_loss": 1.418162614107132
+      },
+      "val": {
+        "bc_loss": 0.12586576408810085,
+        "field_effect_loss": 0.00973788216813571,
+        "field_potential_loss": 0.27150332513782716,
+        "field_preference_loss": 0.5128820389509201,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19965548978911507,
+        "proposal_loss": 4.19468930582904e-06,
+        "rank_acc": 0.8269888758659363,
+        "rank_loss": 0.5128534386555353,
+        "regret_mae": 0.3844732675287459,
+        "success_accuracy": 0.8682900700304244,
+        "total_loss": 1.5073914064301386
+      }
+    },
+    {
+      "epoch": 15,
+      "train": {
+        "bc_loss": 0.10906022026514013,
+        "field_effect_loss": 0.00951166016360124,
+        "field_potential_loss": 0.24163588198522726,
+        "field_preference_loss": 0.507325905478663,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18753065023985174,
+        "proposal_loss": 4.242040543229248e-06,
+        "rank_acc": 0.8380672095550431,
+        "rank_loss": 0.5073215613762537,
+        "regret_mae": 0.37944415625598693,
+        "success_accuracy": 0.8788394770688481,
+        "total_loss": 1.38467841843764
+      },
+      "val": {
+        "bc_loss": 0.12407772118846576,
+        "field_effect_loss": 0.009726297187929353,
+        "field_potential_loss": 0.269621810151471,
+        "field_preference_loss": 0.5113451994127698,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19931338975826898,
+        "proposal_loss": 3.818626560637818e-06,
+        "rank_acc": 0.8303633001115587,
+        "rank_loss": 0.51130553914441,
+        "regret_mae": 0.3830910126368205,
+        "success_accuracy": 0.8681745628515879,
+        "total_loss": 1.496058166027069
+      }
+    },
+    {
+      "epoch": 16,
+      "train": {
+        "bc_loss": 0.10411966954254442,
+        "field_effect_loss": 0.009465229173656553,
+        "field_potential_loss": 0.23694162112143305,
+        "field_preference_loss": 0.5057853725221422,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18760031875636843,
+        "proposal_loss": 3.929118943485972e-06,
+        "rank_acc": 0.8406881408558952,
+        "rank_loss": 0.5057824957701895,
+        "regret_mae": 0.3666251103083293,
+        "success_accuracy": 0.8780882747636901,
+        "total_loss": 1.35282601416111
+      },
+      "val": {
+        "bc_loss": 0.12592858490016726,
+        "field_effect_loss": 0.009718596029819714,
+        "field_potential_loss": 0.270538995663325,
+        "field_preference_loss": 0.5140777445501752,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19766610778040355,
+        "proposal_loss": 3.938984744106872e-06,
+        "rank_acc": 0.825645794471105,
+        "rank_loss": 0.5140392366382811,
+        "regret_mae": 0.40488425890604657,
+        "success_accuracy": 0.8710027436415354,
+        "total_loss": 1.5084967679447598
+      }
+    },
+    {
+      "epoch": 17,
+      "train": {
+        "bc_loss": 0.10320900551353891,
+        "field_effect_loss": 0.009437787695787847,
+        "field_potential_loss": 0.23399021331634787,
+        "field_preference_loss": 0.5059004198345873,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18673738464713097,
+        "proposal_loss": 3.7513855362350214e-06,
+        "rank_acc": 0.8406223000751601,
+        "rank_loss": 0.5058916128344007,
+        "regret_mae": 0.3665856977717744,
+        "success_accuracy": 0.8793778386380937,
+        "total_loss": 1.34524797482623
+      },
+      "val": {
+        "bc_loss": 0.12635401098264587,
+        "field_effect_loss": 0.009661436960515048,
+        "field_potential_loss": 0.26214544143941665,
+        "field_preference_loss": 0.5101073284943899,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1927518852882915,
+        "proposal_loss": 3.5462989116139297e-06,
+        "rank_acc": 0.8336138791508145,
+        "rank_loss": 0.5100681417518191,
+        "regret_mae": 0.3382937941286299,
+        "success_accuracy": 0.8694871465365092,
+        "total_loss": 1.4975009229448106
+      }
+    },
+    {
+      "epoch": 18,
+      "train": {
+        "bc_loss": 0.10179412018300758,
+        "field_effect_loss": 0.00945210562268686,
+        "field_potential_loss": 0.23569712373945448,
+        "field_preference_loss": 0.504312543819348,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1870238735444016,
+        "proposal_loss": 3.593016891222255e-06,
+        "rank_acc": 0.8435247192780176,
+        "rank_loss": 0.5043025161657069,
+        "regret_mae": 0.36600661070810425,
+        "success_accuracy": 0.8775999935136901,
+        "total_loss": 1.338515556520886
+      },
+      "val": {
+        "bc_loss": 0.11730629122919506,
+        "field_effect_loss": 0.009723822896679243,
+        "field_potential_loss": 0.2668940963016616,
+        "field_preference_loss": 0.5103845496972402,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20118692682849038,
+        "proposal_loss": 3.335081335838671e-06,
+        "rank_acc": 0.8340668049123552,
+        "rank_loss": 0.5103463547097312,
+        "regret_mae": 0.3971967614359326,
+        "success_accuracy": 0.8674185143576728,
+        "total_loss": 1.4575166702270508
+      }
+    },
+    {
+      "epoch": 19,
+      "train": {
+        "bc_loss": 0.09701888366705841,
+        "field_effect_loss": 0.00939578995005124,
+        "field_potential_loss": 0.2303258226149612,
+        "field_preference_loss": 0.5027689809600512,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1874675690713856,
+        "proposal_loss": 3.7123108490454292e-06,
+        "rank_acc": 0.8450593766238954,
+        "rank_loss": 0.5027664895686839,
+        "regret_mae": 0.3651750832796097,
+        "success_accuracy": 0.8806694886750646,
+        "total_loss": 1.3066745052735012
+      },
+      "val": {
+        "bc_loss": 0.12064916764696439,
+        "field_effect_loss": 0.009789147310786776,
+        "field_potential_loss": 0.26618921922312844,
+        "field_preference_loss": 0.5258636573950449,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.2000309137834443,
+        "proposal_loss": 3.721368317302323e-06,
+        "rank_acc": 0.8275189035468631,
+        "rank_loss": 0.525830497344335,
+        "regret_mae": 0.40680553515752155,
+        "success_accuracy": 0.8714437716537051,
+        "total_loss": 1.4926239649454753
+      }
+    },
+    {
+      "epoch": 20,
+      "train": {
+        "bc_loss": 0.09741259795717067,
+        "field_effect_loss": 0.009389063136445152,
+        "field_potential_loss": 0.22990970748166242,
+        "field_preference_loss": 0.5037354226741526,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18737832539611393,
+        "proposal_loss": 3.504368060526354e-06,
+        "rank_acc": 0.8435793783929613,
+        "rank_loss": 0.5037313819759421,
+        "regret_mae": 0.3614812975542413,
+        "success_accuracy": 0.8808572888374329,
+        "total_loss": 1.3096330381102033
+      },
+      "val": {
+        "bc_loss": 0.12551961839199066,
+        "field_effect_loss": 0.00964144974326094,
+        "field_potential_loss": 0.2702377488215764,
+        "field_preference_loss": 0.5053084608581331,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19806578920947182,
+        "proposal_loss": 3.535541976715548e-06,
+        "rank_acc": 0.8323609862062666,
+        "rank_loss": 0.5052768604622947,
+        "regret_mae": 0.3880985329548518,
+        "success_accuracy": 0.872850861814287,
+        "total_loss": 1.495032959514194
+      }
+    },
+    {
+      "epoch": 21,
+      "train": {
+        "bc_loss": 0.09099445849036177,
+        "field_effect_loss": 0.009357724558665521,
+        "field_potential_loss": 0.22679272873534095,
+        "field_preference_loss": 0.5019592237141397,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18690396410723528,
+        "proposal_loss": 3.5090738745364233e-06,
+        "rank_acc": 0.8459717482328415,
+        "rank_loss": 0.5019521419372823,
+        "regret_mae": 0.3569582721425427,
+        "success_accuracy": 0.8819194046987427,
+        "total_loss": 1.2719225395056937
+      },
+      "val": {
+        "bc_loss": 0.1169213346309132,
+        "field_effect_loss": 0.009632163474129306,
+        "field_potential_loss": 0.25691480023993385,
+        "field_preference_loss": 0.5171228249867758,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20181949850585726,
+        "proposal_loss": 3.3765298894650186e-06,
+        "rank_acc": 0.8330071502261691,
+        "rank_loss": 0.5170888205369314,
+        "regret_mae": 0.37757547034157646,
+        "success_accuracy": 0.870894236697091,
+        "total_loss": 1.4526626269022624
+      }
+    },
+    {
+      "epoch": 22,
+      "train": {
+        "bc_loss": 0.08927971005646719,
+        "field_effect_loss": 0.00933524470181308,
+        "field_potential_loss": 0.22429877126382458,
+        "field_preference_loss": 0.5003860402438376,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18698722124099731,
+        "proposal_loss": 3.1692379359861257e-06,
+        "rank_acc": 0.8492581099271774,
+        "rank_loss": 0.5003784957031409,
+        "regret_mae": 0.36125429885254967,
+        "success_accuracy": 0.8804357813464271,
+        "total_loss": 1.2588200958238707
+      },
+      "val": {
+        "bc_loss": 0.11955396003193325,
+        "field_effect_loss": 0.009644302565397488,
+        "field_potential_loss": 0.2834768013821708,
+        "field_preference_loss": 0.49570071200529736,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19657663835419548,
+        "proposal_loss": 2.9294760805795604e-06,
+        "rank_acc": 0.8354229165448083,
+        "rank_loss": 0.49566515121195054,
+        "regret_mae": 0.4048614584737354,
+        "success_accuracy": 0.8704707092709012,
+        "total_loss": 1.4683579471376207
+      }
+    },
+    {
+      "epoch": 23,
+      "train": {
+        "bc_loss": 0.08849993182553186,
+        "field_effect_loss": 0.00931584347401642,
+        "field_potential_loss": 0.22267425991594791,
+        "field_preference_loss": 0.49934470115436447,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18664432399802738,
+        "proposal_loss": 3.2455593049639397e-06,
+        "rank_acc": 0.8499041083786223,
+        "rank_loss": 0.4993400391605165,
+        "regret_mae": 0.360809409370025,
+        "success_accuracy": 0.8820842521058189,
+        "total_loss": 1.2518778476450179
+      },
+      "val": {
+        "bc_loss": 0.11901654158201483,
+        "field_effect_loss": 0.009661491753326522,
+        "field_potential_loss": 0.2667437907722261,
+        "field_preference_loss": 0.5112331277794309,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19928998748461405,
+        "proposal_loss": 3.1818816902765925e-06,
+        "rank_acc": 0.8352555632591248,
+        "rank_loss": 0.5112065821886063,
+        "regret_mae": 0.38188037607404923,
+        "success_accuracy": 0.8688396049870385,
+        "total_loss": 1.466876573032803
+      }
+    },
+    {
+      "epoch": 24,
+      "train": {
+        "bc_loss": 0.08692717055479686,
+        "field_effect_loss": 0.009318651838435067,
+        "field_potential_loss": 0.22150142966873115,
+        "field_preference_loss": 0.4990277994010184,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1859870815856589,
+        "proposal_loss": 3.34907730664074e-06,
+        "rank_acc": 0.8512972634699609,
+        "rank_loss": 0.4990229759779241,
+        "regret_mae": 0.35918504289454883,
+        "success_accuracy": 0.882601746254497,
+        "total_loss": 1.2424229085445404
+      },
+      "val": {
+        "bc_loss": 0.11952484688825077,
+        "field_effect_loss": 0.00975932791415188,
+        "field_potential_loss": 0.29336702823638916,
+        "field_preference_loss": 0.49539560576279956,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19757575458950466,
+        "proposal_loss": 3.190176976709659e-06,
+        "rank_acc": 0.8348726597097185,
+        "rank_loss": 0.495365912715594,
+        "regret_mae": 0.4033497969309489,
+        "success_accuracy": 0.8672085007031759,
+        "total_loss": 1.478934771484799
+      }
+    },
+    {
+      "epoch": 25,
+      "train": {
+        "bc_loss": 0.08322953432798386,
+        "field_effect_loss": 0.009336332389567461,
+        "field_potential_loss": 0.22160457364386982,
+        "field_preference_loss": 0.49833980202674866,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18608680988351503,
+        "proposal_loss": 3.299115955239055e-06,
+        "rank_acc": 0.8493981659412384,
+        "rank_loss": 0.4983361214399338,
+        "regret_mae": 0.36144936374492115,
+        "success_accuracy": 0.8815250239438481,
+        "total_loss": 1.2233080284463034
+      },
+      "val": {
+        "bc_loss": 0.11608790689044529,
+        "field_effect_loss": 0.009648644184279773,
+        "field_potential_loss": 0.26099096486965817,
+        "field_preference_loss": 0.5148097376028696,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.2005537384086185,
+        "proposal_loss": 3.0053699902055087e-06,
+        "rank_acc": 0.8350442085001204,
+        "rank_loss": 0.5147705574830373,
+        "regret_mae": 0.398798911107911,
+        "success_accuracy": 0.8704637090365092,
+        "total_loss": 1.4506591426001654
+      }
+    },
+    {
+      "epoch": 26,
+      "train": {
+        "bc_loss": 0.08187239648153384,
+        "field_effect_loss": 0.00927150512062427,
+        "field_potential_loss": 0.21644038375880983,
+        "field_preference_loss": 0.4979762252834108,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18622561275131172,
+        "proposal_loss": 3.219653134717899e-06,
+        "rank_acc": 0.8523989427420828,
+        "rank_loss": 0.497968347536193,
+        "regret_mae": 0.35413394164707923,
+        "success_accuracy": 0.881716997259193,
+        "total_loss": 1.2102421778771613
+      },
+      "val": {
+        "bc_loss": 0.11851631022161907,
+        "field_effect_loss": 0.009645341921390759,
+        "field_potential_loss": 0.2719794437289238,
+        "field_preference_loss": 0.5063139382335875,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19914171679152382,
+        "proposal_loss": 3.0868297547688903e-06,
+        "rank_acc": 0.8335367573632134,
+        "rank_loss": 0.5062821209430695,
+        "regret_mae": 0.3837410658597946,
+        "success_accuracy": 0.8698056671354506,
+        "total_loss": 1.4640845590167575
+      }
+    },
+    {
+      "epoch": 27,
+      "train": {
+        "bc_loss": 0.0785721434901158,
+        "field_effect_loss": 0.009254878723166056,
+        "field_potential_loss": 0.21351627115574148,
+        "field_preference_loss": 0.49621183590756524,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18594955715040365,
+        "proposal_loss": 3.283139847098937e-06,
+        "rank_acc": 0.8532398376199934,
+        "rank_loss": 0.4961962240437667,
+        "regret_mae": 0.35680467304256225,
+        "success_accuracy": 0.8828438007169299,
+        "total_loss": 1.1888496859206095
+      },
+      "val": {
+        "bc_loss": 0.11886513522929615,
+        "field_effect_loss": 0.009609257000394993,
+        "field_potential_loss": 0.2716839959224065,
+        "field_preference_loss": 0.5070300532711877,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20061514029900232,
+        "proposal_loss": 2.8798563612831963e-06,
+        "rank_acc": 0.8314109709527757,
+        "rank_loss": 0.5069955438375473,
+        "regret_mae": 0.3811337600151698,
+        "success_accuracy": 0.8650243613455031,
+        "total_loss": 1.4656231270896063
+      }
+    },
+    {
+      "epoch": 28,
+      "train": {
+        "bc_loss": 0.07702523738973671,
+        "field_effect_loss": 0.009240014679057317,
+        "field_potential_loss": 0.21379283194740614,
+        "field_preference_loss": 0.4957580069700877,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18395106453034613,
+        "proposal_loss": 2.9368364190910747e-06,
+        "rank_acc": 0.8533708370394177,
+        "rank_loss": 0.49574923598104054,
+        "regret_mae": 0.35179730960064465,
+        "success_accuracy": 0.8825683602028422,
+        "total_loss": 1.180778515835603
+      },
+      "val": {
+        "bc_loss": 0.12428568758898312,
+        "field_effect_loss": 0.009715685155242682,
+        "field_potential_loss": 0.2694775735338529,
+        "field_preference_loss": 0.5112031549215317,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20096302860312992,
+        "proposal_loss": 2.7225369320957625e-06,
+        "rank_acc": 0.8368701967928145,
+        "rank_loss": 0.5111701703733869,
+        "regret_mae": 0.39687572916348773,
+        "success_accuracy": 0.8698091672526466,
+        "total_loss": 1.495874007542928
+      }
+    },
+    {
+      "epoch": 29,
+      "train": {
+        "bc_loss": 0.07933070184662938,
+        "field_effect_loss": 0.009223786531947553,
+        "field_potential_loss": 0.21155596586565176,
+        "field_preference_loss": 0.4957353075345357,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18475504550668928,
+        "proposal_loss": 3.1274758877467925e-06,
+        "rank_acc": 0.8538068648841646,
+        "rank_loss": 0.49572674309213954,
+        "regret_mae": 0.3578229778342777,
+        "success_accuracy": 0.8831985344489416,
+        "total_loss": 1.1897377247611682
+      },
+      "val": {
+        "bc_loss": 0.11538442307048374,
+        "field_effect_loss": 0.00973979427686168,
+        "field_potential_loss": 0.2751605494154824,
+        "field_preference_loss": 0.5092001971271303,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.198434103694227,
+        "proposal_loss": 3.109688893548284e-06,
+        "rank_acc": 0.833484646346834,
+        "rank_loss": 0.5091651446289487,
+        "regret_mae": 0.3825343433353636,
+        "success_accuracy": 0.8670964936415354,
+        "total_loss": 1.4539120992024739
+      }
+    },
+    {
+      "epoch": 30,
+      "train": {
+        "bc_loss": 0.07532894016346997,
+        "field_effect_loss": 0.009238159199917896,
+        "field_potential_loss": 0.21124400198459625,
+        "field_preference_loss": 0.4948255568742752,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18387275747954845,
+        "proposal_loss": 3.199564086268462e-06,
+        "rank_acc": 0.8563462942838669,
+        "rank_loss": 0.494819938722584,
+        "regret_mae": 0.35230395363436806,
+        "success_accuracy": 0.8844651447402107,
+        "total_loss": 1.1684001700745688
+      },
+      "val": {
+        "bc_loss": 0.11838398459884855,
+        "field_effect_loss": 0.009639084054571059,
+        "field_potential_loss": 0.27121952424446744,
+        "field_preference_loss": 0.5081077996227477,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19939489414294562,
+        "proposal_loss": 2.992714661963368e-06,
+        "rank_acc": 0.829648945066664,
+        "rank_loss": 0.508072919315762,
+        "regret_mae": 0.3837554454803467,
+        "success_accuracy": 0.8703552020920647,
+        "total_loss": 1.4638316498862372
+      }
+    },
+    {
+      "epoch": 31,
+      "train": {
+        "bc_loss": 0.07409677908031477,
+        "field_effect_loss": 0.009219084678786911,
+        "field_potential_loss": 0.20782110653817654,
+        "field_preference_loss": 0.49436977091762757,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18481638261841404,
+        "proposal_loss": 2.9380520913946384e-06,
+        "rank_acc": 0.8569587543606758,
+        "rank_loss": 0.49436326945821446,
+        "regret_mae": 0.35238729127579266,
+        "success_accuracy": 0.8841354507539008,
+        "total_loss": 1.1579326134588983
+      },
+      "val": {
+        "bc_loss": 0.11740237764186329,
+        "field_effect_loss": 0.009628279315721657,
+        "field_potential_loss": 0.27409548146857154,
+        "field_preference_loss": 0.5002319051159753,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19626346561643812,
+        "proposal_loss": 2.5106690928724775e-06,
+        "rank_acc": 0.8372291227181753,
+        "rank_loss": 0.5001979172229767,
+        "regret_mae": 0.38190865682231057,
+        "success_accuracy": 0.8718812995486789,
+        "total_loss": 1.4539916515350342
+      }
+    },
+    {
+      "epoch": 32,
+      "train": {
+        "bc_loss": 0.07604964046428601,
+        "field_effect_loss": 0.009214523798113482,
+        "field_potential_loss": 0.20773968141939905,
+        "field_preference_loss": 0.4920158146156205,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18329581225083935,
+        "proposal_loss": 2.9126381156806424e-06,
+        "rank_acc": 0.858152251276705,
+        "rank_loss": 0.4920101637641589,
+        "regret_mae": 0.35112110815114445,
+        "success_accuracy": 0.8857296680410703,
+        "total_loss": 1.1651166876157124
+      },
+      "val": {
+        "bc_loss": 0.12121279930902852,
+        "field_effect_loss": 0.009730161943783363,
+        "field_potential_loss": 0.28085361917813617,
+        "field_preference_loss": 0.5057980898353789,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19475418246454662,
+        "proposal_loss": 2.9303708717836544e-06,
+        "rank_acc": 0.8358149098025428,
+        "rank_loss": 0.5057701551251941,
+        "regret_mae": 0.3761419902245204,
+        "success_accuracy": 0.8701381882031759,
+        "total_loss": 1.4845760133531358
+      }
+    },
+    {
+      "epoch": 33,
+      "train": {
+        "bc_loss": 0.07139123782205085,
+        "field_effect_loss": 0.009189292342246821,
+        "field_potential_loss": 0.20229963275293508,
+        "field_preference_loss": 0.49226748074094456,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1826863499979178,
+        "proposal_loss": 3.0383943491566847e-06,
+        "rank_acc": 0.8587411517898241,
+        "rank_loss": 0.49225759050912327,
+        "regret_mae": 0.34416447000371087,
+        "success_accuracy": 0.8848803920878304,
+        "total_loss": 1.1361416222320662
+      },
+      "val": {
+        "bc_loss": 0.11920727272000578,
+        "field_effect_loss": 0.00966562555792431,
+        "field_potential_loss": 0.26555019203159547,
+        "field_preference_loss": 0.5110565589533912,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19725562797652352,
+        "proposal_loss": 2.8034876196316973e-06,
+        "rank_acc": 0.8324535489082336,
+        "rank_loss": 0.5110252896944681,
+        "regret_mae": 0.3725450850195355,
+        "success_accuracy": 0.8703552020920647,
+        "total_loss": 1.4659215874142117
+      }
+    },
+    {
+      "epoch": 34,
+      "train": {
+        "bc_loss": 0.06927861948497593,
+        "field_effect_loss": 0.009127410853074657,
+        "field_potential_loss": 0.1977458138846689,
+        "field_preference_loss": 0.4904492824441857,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18456556730800205,
+        "proposal_loss": 3.1006255483134737e-06,
+        "rank_acc": 0.8611800952090157,
+        "rank_loss": 0.4904416700204213,
+        "regret_mae": 0.35003444821470314,
+        "success_accuracy": 0.8850994929671288,
+        "total_loss": 1.118620277278953
+      },
+      "val": {
+        "bc_loss": 0.12176047182745403,
+        "field_effect_loss": 0.0097102380823344,
+        "field_potential_loss": 0.26769068092107773,
+        "field_preference_loss": 0.5129028641515307,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20022562477323744,
+        "proposal_loss": 2.8636054973200467e-06,
+        "rank_acc": 0.8355740772353278,
+        "rank_loss": 0.5128792574008306,
+        "regret_mae": 0.3866352058119244,
+        "success_accuracy": 0.8727458549870385,
+        "total_loss": 1.4831478993097942
+      }
+    },
+    {
+      "epoch": 35,
+      "train": {
+        "bc_loss": 0.06867818834467067,
+        "field_effect_loss": 0.009153748513199389,
+        "field_potential_loss": 0.20034630513853496,
+        "field_preference_loss": 0.4914962877002027,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18415729701519012,
+        "proposal_loss": 2.787047126023228e-06,
+        "rank_acc": 0.859567816886637,
+        "rank_loss": 0.49148442182275986,
+        "regret_mae": 0.3484601577123006,
+        "success_accuracy": 0.8861720421248012,
+        "total_loss": 1.1202280703518126
+      },
+      "val": {
+        "bc_loss": 0.1208335128095415,
+        "field_effect_loss": 0.009712478559878137,
+        "field_potential_loss": 0.2843129129873382,
+        "field_preference_loss": 0.49648712906572556,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1911649646030532,
+        "proposal_loss": 2.6433810818262928e-06,
+        "rank_acc": 0.8309358722633786,
+        "rank_loss": 0.4964521461062961,
+        "regret_mae": 0.3881209062205421,
+        "success_accuracy": 0.8720948133203719,
+        "total_loss": 1.476528333293067
+      }
+    },
+    {
+      "epoch": 36,
+      "train": {
+        "bc_loss": 0.06601278108751608,
+        "field_effect_loss": 0.009130232973903831,
+        "field_potential_loss": 0.19903298798534605,
+        "field_preference_loss": 0.4901214407549964,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18303600181308058,
+        "proposal_loss": 2.681260509815628e-06,
+        "rank_acc": 0.8609599818785986,
+        "rank_loss": 0.4901083517405722,
+        "regret_mae": 0.34927410177058643,
+        "success_accuracy": 0.8874741254581345,
+        "total_loss": 1.103598083058993
+      },
+      "val": {
+        "bc_loss": 0.12033065201507674,
+        "field_effect_loss": 0.009693974784264961,
+        "field_potential_loss": 0.2807553816172812,
+        "field_preference_loss": 0.5045352131128311,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19831935730245379,
+        "proposal_loss": 2.575074202266276e-06,
+        "rank_acc": 0.8338845637109544,
+        "rank_loss": 0.5044972946246465,
+        "regret_mae": 0.3931685272190306,
+        "success_accuracy": 0.8711182508203719,
+        "total_loss": 1.4804321461253696
+      }
+    },
+    {
+      "epoch": 37,
+      "train": {
+        "bc_loss": 0.06544662435125145,
+        "field_effect_loss": 0.009130532407046607,
+        "field_potential_loss": 0.1940766881323523,
+        "field_preference_loss": 0.48795640137460494,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18330075860851341,
+        "proposal_loss": 2.6914577391481644e-06,
+        "rank_acc": 0.8621632870700624,
+        "rank_loss": 0.4879484768542979,
+        "regret_mae": 0.35001011068622273,
+        "success_accuracy": 0.8873426657583978,
+        "total_loss": 1.092974798546897
+      },
+      "val": {
+        "bc_loss": 0.12463546089000171,
+        "field_effect_loss": 0.009703374804101057,
+        "field_potential_loss": 0.28213178283638424,
+        "field_preference_loss": 0.49967628717422485,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.192433786061075,
+        "proposal_loss": 3.0171427548945453e-06,
+        "rank_acc": 0.8366204301516215,
+        "rank_loss": 0.4996391153997845,
+        "regret_mae": 0.37723280158307815,
+        "success_accuracy": 0.8679505487283071,
+        "total_loss": 1.4962703188260396
+      }
+    },
+    {
+      "epoch": 38,
+      "train": {
+        "bc_loss": 0.06317750242952672,
+        "field_effect_loss": 0.00908253415642927,
+        "field_potential_loss": 0.1925809621396992,
+        "field_preference_loss": 0.4891105244557063,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18155857258372837,
+        "proposal_loss": 2.980336290647756e-06,
+        "rank_acc": 0.8636265330844455,
+        "rank_loss": 0.48910044837329125,
+        "regret_mae": 0.3418588808013333,
+        "success_accuracy": 0.8892269300089942,
+        "total_loss": 1.0811160397198465
+      },
+      "val": {
+        "bc_loss": 0.11953592941992813,
+        "field_effect_loss": 0.009692925313073728,
+        "field_potential_loss": 0.27746040042903686,
+        "field_preference_loss": 0.5079074717230267,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19880394885937372,
+        "proposal_loss": 2.909564753079192e-06,
+        "rank_acc": 0.8294563591480255,
+        "rank_loss": 0.5078650613625845,
+        "regret_mae": 0.39930935038460624,
+        "success_accuracy": 0.8712372581164042,
+        "total_loss": 1.475209845436944
+      }
+    },
+    {
+      "epoch": 39,
+      "train": {
+        "bc_loss": 0.06290878976384799,
+        "field_effect_loss": 0.009090496388833143,
+        "field_potential_loss": 0.190411865297291,
+        "field_preference_loss": 0.48740384355187416,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1831119145370192,
+        "proposal_loss": 2.7904001424556354e-06,
+        "rank_acc": 0.8639598513642947,
+        "rank_loss": 0.4873949276904265,
+        "regret_mae": 0.34694703502787483,
+        "success_accuracy": 0.889193543129497,
+        "total_loss": 1.07566503683726
+      },
+      "val": {
+        "bc_loss": 0.12421774450275633,
+        "field_effect_loss": 0.009799131580318013,
+        "field_potential_loss": 0.290381779273351,
+        "field_preference_loss": 0.5155128439267477,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.205481113659011,
+        "proposal_loss": 2.8464038577011606e-06,
+        "rank_acc": 0.8264558083481259,
+        "rank_loss": 0.5154585540294647,
+        "regret_mae": 0.4111667788690991,
+        "success_accuracy": 0.8696026537153456,
+        "total_loss": 1.521372205681271
+      }
+    },
+    {
+      "epoch": 40,
+      "train": {
+        "bc_loss": 0.06338706700545219,
+        "field_effect_loss": 0.009129570997578816,
+        "field_potential_loss": 0.19314033144878018,
+        "field_preference_loss": 0.4881250812775559,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18214396987524298,
+        "proposal_loss": 2.8074471490274136e-06,
+        "rank_acc": 0.8638640584217178,
+        "rank_loss": 0.4881161306467321,
+        "regret_mae": 0.3490567966881726,
+        "success_accuracy": 0.8893813441197077,
+        "total_loss": 1.0818796538644366
+      },
+      "val": {
+        "bc_loss": 0.12153194492889775,
+        "field_effect_loss": 0.00967243745819562,
+        "field_potential_loss": 0.2685299176308844,
+        "field_preference_loss": 0.5152279519372516,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20035429216093487,
+        "proposal_loss": 2.6729868523034384e-06,
+        "rank_acc": 0.832026332616806,
+        "rank_loss": 0.5151940865649117,
+        "regret_mae": 0.3840869946612252,
+        "success_accuracy": 0.8719793061415354,
+        "total_loss": 1.4849157200919256
+      }
+    },
+    {
+      "epoch": 41,
+      "train": {
+        "bc_loss": 0.06239371741604474,
+        "field_effect_loss": 0.009085566154681146,
+        "field_potential_loss": 0.1902465485036373,
+        "field_preference_loss": 0.48685900246103603,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18294654517538017,
+        "proposal_loss": 2.722238463093163e-06,
+        "rank_acc": 0.865128127237161,
+        "rank_loss": 0.48684963087240857,
+        "regret_mae": 0.3439733439849483,
+        "success_accuracy": 0.8892582298980819,
+        "total_loss": 1.0724658626649115
+      },
+      "val": {
+        "bc_loss": 0.12730718817975786,
+        "field_effect_loss": 0.009778164042573836,
+        "field_potential_loss": 0.2869192063808441,
+        "field_preference_loss": 0.5146783143281937,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20081902626487944,
+        "proposal_loss": 3.2313589725971624e-06,
+        "rank_acc": 0.8267917864852481,
+        "rank_loss": 0.5146431111627154,
+        "regret_mae": 0.3783157186375724,
+        "success_accuracy": 0.8669634858767191,
+        "total_loss": 1.5330877900123596
+      }
+    },
+    {
+      "epoch": 42,
+      "train": {
+        "bc_loss": 0.0626623068884429,
+        "field_effect_loss": 0.0090403676020085,
+        "field_potential_loss": 0.1842424590140581,
+        "field_preference_loss": 0.4857383800877465,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18264496905936134,
+        "proposal_loss": 2.8586492027847676e-06,
+        "rank_acc": 0.8663867571287684,
+        "rank_loss": 0.485730672462119,
+        "regret_mae": 0.3432990399499734,
+        "success_accuracy": 0.8889264497492049,
+        "total_loss": 1.0661254309945636
+      },
+      "val": {
+        "bc_loss": 0.12408970793088277,
+        "field_effect_loss": 0.009791983369116982,
+        "field_potential_loss": 0.2884411397907469,
+        "field_preference_loss": 0.5070943203237321,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19478629032770792,
+        "proposal_loss": 2.612230711444378e-06,
+        "rank_acc": 0.8306023412280612,
+        "rank_loss": 0.5070371064874861,
+        "regret_mae": 0.39188093774848515,
+        "success_accuracy": 0.8709992435243394,
+        "total_loss": 1.5096297396553888
+      }
+    },
+    {
+      "epoch": 43,
+      "train": {
+        "bc_loss": 0.06061812096999751,
+        "field_effect_loss": 0.009025660666844083,
+        "field_potential_loss": 0.18237425883611044,
+        "field_preference_loss": 0.4840524382889271,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1814773807095157,
+        "proposal_loss": 2.63972543892679e-06,
+        "rank_acc": 0.8683239453368716,
+        "rank_loss": 0.4840383720066812,
+        "regret_mae": 0.34438371782501537,
+        "success_accuracy": 0.891520183119509,
+        "total_loss": 1.0518468999200397
+      },
+      "val": {
+        "bc_loss": 0.11816486426525646,
+        "field_effect_loss": 0.009715757746663358,
+        "field_potential_loss": 0.2728041427003013,
+        "field_preference_loss": 0.5134753336509069,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19647487501303354,
+        "proposal_loss": 2.819993268430052e-06,
+        "rank_acc": 0.8295322292380862,
+        "rank_loss": 0.5134350508451462,
+        "regret_mae": 0.36835627092255485,
+        "success_accuracy": 0.8726373480425941,
+        "total_loss": 1.4705909887949626
+      }
+    },
+    {
+      "epoch": 44,
+      "train": {
+        "bc_loss": 0.05858887419001096,
+        "field_effect_loss": 0.009014137302680561,
+        "field_potential_loss": 0.17998419205347696,
+        "field_preference_loss": 0.4837602886060874,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18084371027847132,
+        "proposal_loss": 2.7783545861994853e-06,
+        "rank_acc": 0.8685827056566874,
+        "rank_loss": 0.4837554353806708,
+        "regret_mae": 0.3393647347887357,
+        "success_accuracy": 0.8933668873376317,
+        "total_loss": 1.0387242808938026
+      },
+      "val": {
+        "bc_loss": 0.12358489135901134,
+        "field_effect_loss": 0.009757014431266321,
+        "field_potential_loss": 0.27753130926026237,
+        "field_preference_loss": 0.5106030586693022,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.2019328855805927,
+        "proposal_loss": 2.872162629804128e-06,
+        "rank_acc": 0.8313619726234012,
+        "rank_loss": 0.5105618288119634,
+        "regret_mae": 0.3935813324318992,
+        "success_accuracy": 0.8719688057899475,
+        "total_loss": 1.4987553225623236
+      }
+    },
+    {
+      "epoch": 45,
+      "train": {
+        "bc_loss": 0.06211154584565924,
+        "field_effect_loss": 0.009009629462121261,
+        "field_potential_loss": 0.1797659724122948,
+        "field_preference_loss": 0.4829809359378285,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18154007113642162,
+        "proposal_loss": 2.7243780620109546e-06,
+        "rank_acc": 0.869540173974302,
+        "rank_loss": 0.4829693068232801,
+        "regret_mae": 0.34195822953350014,
+        "success_accuracy": 0.8894961112075381,
+        "total_loss": 1.0554731090863545
+      },
+      "val": {
+        "bc_loss": 0.12311960260073344,
+        "field_effect_loss": 0.009753933254008492,
+        "field_potential_loss": 0.2832062625222736,
+        "field_preference_loss": 0.51215817199813,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1978578261203236,
+        "proposal_loss": 2.6564394678037693e-06,
+        "rank_acc": 0.8262186778916253,
+        "rank_loss": 0.5121352076530457,
+        "regret_mae": 0.3934950464301639,
+        "success_accuracy": 0.8722873263888888,
+        "total_loss": 1.5036307043499417
+      }
+    },
+    {
+      "epoch": 46,
+      "train": {
+        "bc_loss": 0.05837097608794769,
+        "field_effect_loss": 0.00900600483873859,
+        "field_potential_loss": 0.1772993834068378,
+        "field_preference_loss": 0.4828516079319848,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18079392653372553,
+        "proposal_loss": 2.6447014218117856e-06,
+        "rank_acc": 0.8700695170296563,
+        "rank_loss": 0.4828408935831653,
+        "regret_mae": 0.338041996376382,
+        "success_accuracy": 0.8933105477028422,
+        "total_loss": 1.0335509288642142
+      },
+      "val": {
+        "bc_loss": 0.11962136626243591,
+        "field_effect_loss": 0.00977536528888676,
+        "field_potential_loss": 0.2841544474164645,
+        "field_preference_loss": 0.5076187567578422,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19883575373225743,
+        "proposal_loss": 2.8428612368871754e-06,
+        "rank_acc": 0.8272956046793196,
+        "rank_loss": 0.5075751029782825,
+        "regret_mae": 0.3819070971674389,
+        "success_accuracy": 0.8725218408637576,
+        "total_loss": 1.483001470565796
+      }
+    },
+    {
+      "epoch": 47,
+      "train": {
+        "bc_loss": 0.056888495261470474,
+        "field_effect_loss": 0.008996267059248768,
+        "field_potential_loss": 0.17572223974598777,
+        "field_preference_loss": 0.48162176501419807,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1819558512005541,
+        "proposal_loss": 2.7248440422530016e-06,
+        "rank_acc": 0.8706941720512178,
+        "rank_loss": 0.4816102724936273,
+        "regret_mae": 0.33934587695532376,
+        "success_accuracy": 0.8937508347961638,
+        "total_loss": 1.0235132459137175
+      },
+      "val": {
+        "bc_loss": 0.1271992925968435,
+        "field_effect_loss": 0.009742060934917795,
+        "field_potential_loss": 0.2902405063311259,
+        "field_preference_loss": 0.5039413074652354,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.2013954081469112,
+        "proposal_loss": 2.5390799641576855e-06,
+        "rank_acc": 0.830833935075336,
+        "rank_loss": 0.5039093080494139,
+        "regret_mae": 0.42165683541033006,
+        "success_accuracy": 0.8715382781293657,
+        "total_loss": 1.5226119624243841
+      }
+    },
+    {
+      "epoch": 48,
+      "train": {
+        "bc_loss": 0.059761072146809764,
+        "field_effect_loss": 0.008944734704629,
+        "field_potential_loss": 0.16962687795360884,
+        "field_preference_loss": 0.47961554303765297,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18107213100625408,
+        "proposal_loss": 2.7166744681987134e-06,
+        "rank_acc": 0.8729642530282339,
+        "rank_loss": 0.4796089517573516,
+        "regret_mae": 0.3362344424757693,
+        "success_accuracy": 0.8938509954346551,
+        "total_loss": 1.0290098670456145
+      },
+      "val": {
+        "bc_loss": 0.12089842102593845,
+        "field_effect_loss": 0.009804613189771771,
+        "field_potential_loss": 0.2921220304237472,
+        "field_preference_loss": 0.5068873051140044,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.1985766655868954,
+        "proposal_loss": 2.441757273017073e-06,
+        "rank_acc": 0.8255698184172312,
+        "rank_loss": 0.5068462524149153,
+        "regret_mae": 0.39554058346483445,
+        "success_accuracy": 0.8698966734939151,
+        "total_loss": 1.495993024773068
+      }
+    },
+    {
+      "epoch": 49,
+      "train": {
+        "bc_loss": 0.0620340155178888,
+        "field_effect_loss": 0.00897529934041409,
+        "field_potential_loss": 0.1719647501077917,
+        "field_preference_loss": 0.48096829445825684,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.18113424277140033,
+        "proposal_loss": 2.6789852619736444e-06,
+        "rank_acc": 0.8712435265382131,
+        "rank_loss": 0.4809506870806217,
+        "regret_mae": 0.3406081758439541,
+        "success_accuracy": 0.8928598264853159,
+        "total_loss": 1.0447475984692574
+      },
+      "val": {
+        "bc_loss": 0.12231180816888809,
+        "field_effect_loss": 0.00987395434640348,
+        "field_potential_loss": 0.3045791784922282,
+        "field_preference_loss": 0.5076191094186571,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.20019585639238358,
+        "proposal_loss": 2.1593938110224067e-06,
+        "rank_acc": 0.8277939094437493,
+        "rank_loss": 0.507586282160547,
+        "regret_mae": 0.42900553345680237,
+        "success_accuracy": 0.8771071334679922,
+        "total_loss": 1.515727824634976
+      }
+    },
+    {
+      "epoch": 50,
+      "train": {
+        "bc_loss": 0.05517530679288837,
+        "field_effect_loss": 0.008948124046178741,
+        "field_potential_loss": 0.16716017718944284,
+        "field_preference_loss": 0.4785834302504857,
+        "lattice_edges": 3830.0,
+        "progress_mae": 0.1795549239549372,
+        "proposal_loss": 2.5497496791457606e-06,
+        "rank_acc": 0.8750536640485128,
+        "rank_loss": 0.4785781506862905,
+        "regret_mae": 0.3398860345284144,
+        "success_accuracy": 0.8962381482124329,
+        "total_loss": 1.0022349316212866
+      },
+      "val": {
+        "bc_loss": 0.1215214677568939,
+        "field_effect_loss": 0.009849462761647172,
+        "field_potential_loss": 0.2982282191514969,
+        "field_preference_loss": 0.5109110491143333,
+        "lattice_edges": 3833.3333333333335,
+        "progress_mae": 0.19872083846065733,
+        "proposal_loss": 2.441965282034491e-06,
+        "rank_acc": 0.8261403408315446,
+        "rank_loss": 0.5108737051486969,
+        "regret_mae": 0.423651796248224,
+        "success_accuracy": 0.8778596818447113,
+        "total_loss": 1.5098645091056824
+      }
+    }
+  ]
+}