{
  "config": {
    "epochs": 200,
    "batch_size": 8,
    "dataset_size": 64,
    "grid": 64,
    "lr": 0.002,
    "consistency_weight": 1.0,
    "seed": 0,
    "device": "mps",
    "forward_model": "gaussian",
    "forward_sigma_px": 4.5
  },
  "history": [
    0.5567391030490398,
    0.37703684344887733,
    0.30704817548394203,
    0.2513122968375683,
    0.20789939165115356,
    0.17624378390610218,
    0.15377543307840824,
    0.13707710430026054,
    0.12503274250775576,
    0.11610893812030554,
    0.10920202825218439,
    0.10366012062877417,
    0.09944107662886381,
    0.09592678025364876,
    0.09313659928739071,
    0.09068253263831139,
    0.08866022154688835,
    0.08697561640292406,
    0.08550380449742079,
    0.0842436458915472,
    0.08310946729034185,
    0.08210133761167526,
    0.08122950792312622,
    0.08043504785746336,
    0.07979514449834824,
    0.0791627336293459,
    0.0786021277308464,
    0.07812768500298262,
    0.07761684618890285,
    0.07720845751464367,
    0.0768478037789464,
    0.0765686109662056,
    0.07618657406419516,
    0.07593129854649305,
    0.07567358110100031,
    0.07541041728109121,
    0.07518432103097439,
    0.07493862044066191,
    0.07475274614989758,
    0.07456374727189541,
    0.07439965475350618,
    0.07425445504486561,
    0.07408564537763596,
    0.07393893785774708,
    0.07380379270762205,
    0.07371699437499046,
    0.07357585616409779,
    0.07347930036485195,
    0.07337613683193922,
    0.07327591348439455,
    0.07315656077116728,
    0.0731016043573618,
    0.07299685291945934,
    0.07293033879250288,
    0.07284928020089865,
    0.07278184406459332,
    0.0726948045194149,
    0.07264522649347782,
    0.07260565552860498,
    0.07252266444265842,
    0.07248253747820854,
    0.07241528294980526,
    0.0723672267049551,
    0.07233524136245251,
    0.07227082457393408,
    0.07222518604248762,
    0.07219320628792048,
    0.07216380164027214,
    0.07210987340658903,
    0.07208450976759195,
    0.07204032503068447,
    0.07200989127159119,
    0.07197732990607619,
    0.07193727605044842,
    0.07192076463252306,
    0.07189632486552,
    0.07185683958232403,
    0.07183338236063719,
    0.0718119814991951,
    0.07178394217044115,
    0.07175531424582005,
    0.07173179369419813,
    0.07169996295124292,
    0.07168740313500166,
    0.07166620437055826,
    0.07163839414715767,
    0.07162577379494905,
    0.07160577178001404,
    0.07160489074885845,
    0.07158240769058466,
    0.07155606709420681,
    0.0715420925989747,
    0.07153397984802723,
    0.0715121328830719,
    0.0714907106012106,
    0.0714725237339735,
    0.07147086691111326,
    0.07144990283995867,
    0.07144117914140224,
    0.07142881397157907,
    0.07141822669655085,
    0.071401821449399,
    0.07139936741441488,
    0.07137478701770306,
    0.07136495783925056,
    0.0713663725182414,
    0.07135887071490288,
    0.07134600449353456,
    0.07133242394775152,
    0.07132337335497141,
    0.0713175656273961,
    0.07130793109536171,
    0.07129425182938576,
    0.07128519285470247,
    0.07128226570785046,
    0.07127349078655243,
    0.07126602809876204,
    0.07126884814351797,
    0.07125518098473549,
    0.07124917767941952,
    0.07123759668320417,
    0.071230742149055,
    0.07121757883578539,
    0.07122342102229595,
    0.07121550850570202,
    0.07121784146875143,
    0.07119968719780445,
    0.07119478564709425,
    0.07119330018758774,
    0.07118833623826504,
    0.07118134945631027,
    0.07119169738143682,
    0.07116888929158449,
    0.07117053400725126,
    0.07117703091353178,
    0.07115927897393703,
    0.07115941820666194,
    0.07115516811609268,
    0.07115022744983435,
    0.07114393077790737,
    0.0711393104866147,
    0.07114613242447376,
    0.07112972717732191,
    0.07113201078027487,
    0.07113241590559483,
    0.07111924607306719,
    0.07112239301204681,
    0.07111891638487577,
    0.07111316360533237,
    0.07111237850040197,
    0.07111064624041319,
    0.07111744116991758,
    0.07111241295933723,
    0.07111366093158722,
    0.07111791800707579,
    0.07111515756696463,
    0.0711008571088314,
    0.0710973753593862,
    0.07109776418656111,
    0.07109313178807497,
    0.0710864719003439,
    0.07110285758972168,
    0.07109163049608469,
    0.0710831182077527,
    0.07109935116022825,
    0.07108418829739094,
    0.07108188793063164,
    0.07108257338404655,
    0.07108830939978361,
    0.07108316756784916,
    0.07107411697506905,
    0.07107533607631922,
    0.0710767200216651,
    0.0710803447291255,
    0.07107866276055574,
    0.0710811149328947,
    0.07107741292566061,
    0.07107247225940228,
    0.07107514701783657,
    0.07107173278927803,
    0.07107879314571619,
    0.07107033394277096,
    0.0710739092901349,
    0.07108107954263687,
    0.07108086440712214,
    0.07106953207403421,
    0.0710679879412055,
    0.07106881123036146,
    0.07106904312968254,
    0.07106966245919466,
    0.07106941845268011,
    0.07107054069638252,
    0.07106670178472996,
    0.07107471022754908,
    0.07108479831367731,
    0.07108394801616669,
    0.07107358612120152,
    0.0710753770545125,
    0.07106765080243349,
    0.07107784505933523
  ],
  "final_loss": 0.07107784505933523
}