1{2 "node": "m3u96b",3 "chip": "Apple M3 Ultra",4 "torch": "2.15.0",5 "device": "mps",6 "n_origins": 16299,7 "loader": {8 "workers_0": {9 "ex_per_s": 3970.950446598954410 },11 "workers_4": {12 "ex_per_s": 13015.52570396499313 },14 "workers_8": {15 "ex_per_s": 38486.3356428297516 }17 },18 "train": {19 "small_bs32": {20 "params": 3455619,21 "steps_per_s": 20.808967283900596,22 "ex_per_s": 665.8869530848191,23 "peak_mem_mb": 53.906944,24 "loss_finite": true25 },26 "small_bs64": {27 "params": 3455619,28 "steps_per_s": 12.212416590445695,29 "ex_per_s": 781.5946617885245,30 "peak_mem_mb": 68.529152,31 "loss_finite": true32 },33 "small_bs128": {34 "params": 3455619,35 "steps_per_s": 5.925401298878898,36 "ex_per_s": 758.451366256499,37 "peak_mem_mb": 96.797184,38 "loss_finite": true39 },40 "spec_bs32": {41 "params": 7677955,42 "steps_per_s": 11.544136009460699,43 "ex_per_s": 369.41235230274236,44 "peak_mem_mb": 114.085376,45 "loss_finite": true46 },47 "spec_bs64": {48 "params": 7677955,49 "steps_per_s": 6.285685112498215,50 "ex_per_s": 402.28384719988577,51 "peak_mem_mb": 119.197184,52 "loss_finite": true53 },54 "spec_bs128": {55 "params": 7677955,56 "steps_per_s": 2.9673766457169934,57 "ex_per_s": 379.82421065177516,58 "peak_mem_mb": 146.711552,59 "loss_finite": true60 },61 "base_bs32": {62 "params": 20718339,63 "steps_per_s": 6.10752187132323,64 "ex_per_s": 195.44069988234335,65 "peak_mem_mb": 275.288576,66 "loss_finite": true67 },68 "base_bs64": {69 "params": 20718339,70 "steps_per_s": 3.239651990508832,71 "ex_per_s": 207.33772739256526,72 "peak_mem_mb": 283.539968,73 "loss_finite": true74 },75 "base_bs128": {76 "params": 20718339,77 "steps_per_s": 1.5058899942717616,78 "ex_per_s": 192.7539192667855,79 "peak_mem_mb": 315.058176,80 "loss_finite": true81 }82 },83 "inference": {84 "small_bs32": {85 "latency_ms_batch1": 4.309260845184326,86 "latency_ms_per_origin_batched": 0.3443192690610885687 },88 "small_bs64": {89 "latency_ms_batch1": 4.311156272888184,90 "latency_ms_per_origin_batched": 0.254152342677116491 },92 "small_bs128": {93 "latency_ms_batch1": 4.303455352783203,94 "latency_ms_per_origin_batched": 0.2710334956645965695 },96 "spec_bs32": {97 "latency_ms_batch1": 5.084848403930664,98 "latency_ms_per_origin_batched": 0.586701557040214599 },100 "spec_bs64": {101 "latency_ms_batch1": 5.049645900726318,102 "latency_ms_per_origin_batched": 0.4757307469844818103 },104 "spec_bs128": {105 "latency_ms_batch1": 5.115711688995361,106 "latency_ms_per_origin_batched": 0.5341288633644581107 },108 "base_bs32": {109 "latency_ms_batch1": 5.908381938934326,110 "latency_ms_per_origin_batched": 5.037868395447731111 },112 "base_bs64": {113 "latency_ms_batch1": 5.8892011642456055,114 "latency_ms_per_origin_batched": 1.585175283253193115 },116 "base_bs128": {117 "latency_ms_batch1": 5.894708633422852,118 "latency_ms_per_origin_batched": 1.7552338540554047119 }120 }121}