Skip to content

Instantly share code, notes, and snippets.

Show Gist options
  • Save samuela/face71187ce6696a76610cdf543f594f to your computer and use it in GitHub Desktop.
Save samuela/face71187ce6696a76610cdf543f594f to your computer and use it in GitHub Desktop.
/Users/skainswo/dev/jax/jax/lib/xla_bridge.py:116: UserWarning: No GPU/TPU found, falling back to CPU.
warnings.warn('No GPU/TPU found, falling back to CPU.')
opt_cost_fwd = 4.999999046325684, opt_cost_bwd = 2.480832099914551 in 3.0865352153778076s
[3.4355838e-09 1.7177919e-09]
[1.4196247 0.70981234]
l2 error: 0.64887934923172
Episode 0:
excess fwd cost = 4433.6806640625
bwd cost = -0.00036362005630508065
bwd x0 - x0 = [-5.0067902e-06 2.7418137e-06]
fwd xT = [-18.23771 19.609293]
fwd xT norm sq. = 717.1384887695312
elapsed = 9.440975904464722s
Episode 1:
excess fwd cost = 2414.2080078125
bwd cost = -0.003976716194301844
bwd x0 - x0 = [-1.1205673e-05 1.0728836e-05]
fwd xT = [-13.286568 10.83102 ]
fwd xT norm sq. = 293.8438720703125
elapsed = 0.026827096939086914s
Episode 2:
excess fwd cost = 978.572021484375
bwd cost = 0.000850033073220402
bwd x0 - x0 = [ 4.5299530e-06 -4.2915344e-06]
fwd xT = [-5.0759606 -3.0983973]
fwd xT norm sq. = 35.365440368652344
elapsed = 0.03938484191894531s
Episode 3:
excess fwd cost = 638.9349365234375
bwd cost = -0.0014253611443564296
bwd x0 - x0 = [ 9.536743e-07 -1.668930e-06]
fwd xT = [ 3.0668972 -6.535607 ]
fwd xT norm sq. = 52.12001419067383
elapsed = 0.034822702407836914s
Episode 4:
excess fwd cost = 596.6362915039062
bwd cost = -0.00039554378599859774
bwd x0 - x0 = [4.7683716e-07 4.7683716e-07]
fwd xT = [ 6.60776 -7.2236733]
fwd xT norm sq. = 95.84394836425781
elapsed = 0.027399063110351562s
Episode 5:
excess fwd cost = 591.2402954101562
bwd cost = 8.887634612619877e-05
bwd x0 - x0 = [ 2.1457672e-06 -2.4437904e-06]
fwd xT = [ 7.661504 -6.6715755]
fwd xT norm sq. = 103.20855712890625
elapsed = 0.04132676124572754s
Episode 6:
excess fwd cost = 560.5917358398438
bwd cost = 0.0014475849457085133
bwd x0 - x0 = [-2.3841858e-07 -2.0861626e-06]
fwd xT = [ 7.7265925 -5.4349055]
fwd xT norm sq. = 89.23843383789062
elapsed = 0.035909175872802734s
Episode 7:
excess fwd cost = 506.8153991699219
bwd cost = 0.0005397359491325915
bwd x0 - x0 = [ 1.0967255e-05 -1.0788441e-05]
fwd xT = [ 7.2825327 -3.8685467]
fwd xT norm sq. = 68.00093841552734
elapsed = 0.039106130599975586s
Episode 8:
excess fwd cost = 442.6976318359375
bwd cost = 0.00010282648145221174
bwd x0 - x0 = [-2.2649765e-06 1.1920929e-06]
fwd xT = [ 6.524426 -2.29671 ]
fwd xT norm sq. = 47.84300994873047
elapsed = 0.030642032623291016s
Episode 9:
excess fwd cost = 378.64654541015625
bwd cost = 6.115850555943325e-05
bwd x0 - x0 = [ 5.00679e-06 -5.00679e-06]
fwd xT = [ 5.5780153 -0.9483429]
fwd xT norm sq. = 32.01361083984375
elapsed = 0.04029202461242676s
Episode 10:
excess fwd cost = 320.24322509765625
bwd cost = -0.0005017887451685965
bwd x0 - x0 = [-3.337860e-06 4.529953e-06]
fwd xT = [4.5136857 0.09381027]
fwd xT norm sq. = 20.382158279418945
elapsed = 0.029629945755004883s
Episode 11:
excess fwd cost = 269.6427001953125
bwd cost = 0.0001585867430549115
bwd x0 - x0 = [ 3.337860e-06 -3.993511e-06]
fwd xT = [3.3808198 0.84394693]
fwd xT norm sq. = 12.142189025878906
elapsed = 0.03337717056274414s
Episode 12:
excess fwd cost = 227.43576049804688
bwd cost = -0.00013258190301712602
bwd x0 - x0 = [-1.3113022e-06 -1.1920929e-07]
fwd xT = [2.2756245 1.3285081]
fwd xT norm sq. = 6.9434003829956055
elapsed = 0.02678990364074707s
Episode 13:
excess fwd cost = 193.2466278076172
bwd cost = 4.753349276143126e-05
bwd x0 - x0 = [-1.4066696e-05 8.4638596e-06]
fwd xT = [1.3114718 1.5748456]
fwd xT norm sq. = 4.20009708404541
elapsed = 0.044046878814697266s
Episode 14:
excess fwd cost = 165.9762420654297
bwd cost = -7.634679786860943e-05
bwd x0 - x0 = [2.3841858e-07 0.0000000e+00]
fwd xT = [0.5588691 1.6346344]
fwd xT norm sq. = 2.9843642711639404
elapsed = 0.030217885971069336s
Episode 15:
excess fwd cost = 144.2540283203125
bwd cost = 0.00014902766270097345
bwd x0 - x0 = [ 1.3113022e-05 -1.0073185e-05]
fwd xT = [0.0280974 1.5736109]
fwd xT norm sq. = 2.4770407676696777
elapsed = 0.034775733947753906s
Episode 16:
excess fwd cost = 126.82124328613281
bwd cost = 8.222331962315366e-05
bwd x0 - x0 = [-4.7683716e-06 2.8610229e-06]
fwd xT = [-0.31404424 1.4477062 ]
fwd xT norm sq. = 2.194477081298828
elapsed = 0.04455280303955078s
Episode 17:
excess fwd cost = 112.68049621582031
bwd cost = 0.0004169352469034493
bwd x0 - x0 = [ 1.1920929e-06 -3.7550926e-06]
fwd xT = [-0.515464 1.2950497]
fwd xT norm sq. = 1.942856788635254
elapsed = 0.03525519371032715s
Episode 18:
excess fwd cost = 101.0862808227539
bwd cost = 1.947976852534339e-05
bwd x0 - x0 = [-4.2915344e-06 3.6954880e-06]
fwd xT = [-0.62002987 1.1385847 ]
fwd xT norm sq. = 1.680812120437622
elapsed = 0.034853219985961914s
Episode 19:
excess fwd cost = 91.48777770996094
bwd cost = 6.018375279381871e-05
bwd x0 - x0 = [-2.503395e-06 8.344650e-07]
fwd xT = [-0.6613101 0.99063236]
fwd xT norm sq. = 1.4186835289001465
elapsed = 0.03776407241821289s
Episode 20:
excess fwd cost = 83.4735336303711
bwd cost = -7.978006578923669e-06
bwd x0 - x0 = [-5.4836273e-06 3.6954880e-06]
fwd xT = [-0.66295934 0.85684115]
fwd xT norm sq. = 1.1736918687820435
elapsed = 0.03821396827697754s
Episode 21:
excess fwd cost = 76.73058319091797
bwd cost = 9.484091424383223e-05
bwd x0 - x0 = [-6.9141388e-06 3.5762787e-06]
fwd xT = [-0.640877 0.73895824]
fwd xT norm sq. = 0.9567825794219971
elapsed = 0.02253890037536621s
Episode 22:
excess fwd cost = 71.01675415039062
bwd cost = -2.4121774913510308e-05
bwd x0 - x0 = [ 4.7683716e-07 -1.7881393e-07]
fwd xT = [-0.6054702 0.63664865]
fwd xT norm sq. = 0.7719156742095947
elapsed = 0.03909587860107422s
Episode 23:
excess fwd cost = 66.14268493652344
bwd cost = 6.323238631011918e-05
bwd x0 - x0 = [-1.4305115e-06 -1.2516975e-06]
fwd xT = [-0.56339216 0.5485825 ]
fwd xT norm sq. = 0.6183534860610962
elapsed = 0.03980898857116699s
Episode 24:
excess fwd cost = 61.958457946777344
bwd cost = -0.0002041353436652571
bwd x0 - x0 = [ 8.821487e-06 -5.722046e-06]
fwd xT = [-0.51879793 0.4730737 ]
fwd xT norm sq. = 0.4929500222206116
elapsed = 0.042897939682006836s
Episode 25:
excess fwd cost = 58.344905853271484
bwd cost = -0.00010516932525206357
bwd x0 - x0 = [ 1.21593475e-05 -6.73532486e-06]
fwd xT = [-0.47421828 0.40838704]
fwd xT norm sq. = 0.39166295528411865
elapsed = 0.05564594268798828s
Episode 26:
excess fwd cost = 55.20578384399414
bwd cost = 8.375753532163799e-05
bwd x0 - x0 = [-1.2040138e-05 7.5101852e-06]
fwd xT = [-0.43112585 0.35291633]
fwd xT norm sq. = 0.3104194402694702
elapsed = 0.041863203048706055s
Episode 27:
excess fwd cost = 52.4636116027832
bwd cost = 0.0001326190249528736
bwd x0 - x0 = [-1.6331673e-05 8.3446503e-06]
fwd xT = [-0.39032146 0.30524108]
fwd xT norm sq. = 0.24552296102046967
elapsed = 0.030037879943847656s
Episode 28:
excess fwd cost = 50.05569076538086
bwd cost = 5.074110231362283e-05
bwd x0 - x0 = [-4.7683716e-07 1.1920929e-07]
fwd xT = [-0.35218972 0.26413882]
fwd xT norm sq. = 0.19380691647529602
elapsed = 0.03020000457763672s
Episode 29:
excess fwd cost = 47.930381774902344
bwd cost = -3.466169800958596e-05
bwd x0 - x0 = [ 7.1525574e-07 -7.7486038e-07]
fwd xT = [-0.31685194 0.22857407]
fwd xT norm sq. = 0.15264126658439636
elapsed = 0.03129720687866211s
Episode 30:
excess fwd cost = 46.045772552490234
bwd cost = 0.00014050710888113827
bwd x0 - x0 = [-1.7046928e-05 9.1791153e-06]
fwd xT = [-0.2842773 0.19767824]
fwd xT norm sq. = 0.11989026516675949
elapsed = 0.029977798461914062s
Episode 31:
excess fwd cost = 44.36670684814453
bwd cost = -0.00010844337521120906
bwd x0 - x0 = [ 1.4305115e-05 -7.5697899e-06]
fwd xT = [-0.25434336 0.17072964]
fwd xT norm sq. = 0.09383915364742279
elapsed = 0.03408169746398926s
Episode 32:
excess fwd cost = 42.86437225341797
bwd cost = 9.098007467400748e-06
bwd x0 - x0 = [ 2.3841858e-06 -2.3245811e-06]
fwd xT = [-0.22688219 0.14712472]
fwd xT norm sq. = 0.07312121242284775
elapsed = 0.023421049118041992s
Episode 33:
excess fwd cost = 41.51456069946289
bwd cost = -4.038035694975406e-05
bwd x0 - x0 = [-4.6491623e-06 3.6954880e-06]
fwd xT = [-0.2017046 0.12636487]
fwd xT norm sq. = 0.05665282905101776
elapsed = 0.03275299072265625s
Episode 34:
excess fwd cost = 40.29684066772461
bwd cost = -0.00014530781481880695
bwd x0 - x0 = [ 7.8678131e-06 -3.2186508e-06]
fwd xT = [-0.17861669 0.10803004]
fwd xT norm sq. = 0.043574411422014236
elapsed = 0.04041910171508789s
Episode 35:
excess fwd cost = 39.1943359375
bwd cost = 3.0148781661409885e-05
bwd x0 - x0 = [-6.6757202e-06 2.6226044e-06]
fwd xT = [-0.15743008 0.09177222]
fwd xT norm sq. = 0.03320637345314026
elapsed = 0.027966976165771484s
Episode 36:
excess fwd cost = 38.19227600097656
bwd cost = 6.424080766009865e-06
bwd x0 - x0 = [ 4.7683716e-07 -2.3841858e-07]
fwd xT = [-0.13796513 0.07730229]
fwd xT norm sq. = 0.025010021403431892
elapsed = 0.07500720024108887s
Episode 37:
excess fwd cost = 37.27840805053711
bwd cost = 0.00012347858864814043
bwd x0 - x0 = [-1.4305115e-05 6.0796738e-06]
fwd xT = [-0.12005656 0.06437215]
fwd xT norm sq. = 0.01855735294520855
elapsed = 0.04473590850830078s
Episode 38:
excess fwd cost = 36.442138671875
bwd cost = -4.1324241465190426e-05
bwd x0 - x0 = [9.5367432e-07 2.3841858e-07]
fwd xT = [-0.10355197 0.05277927]
fwd xT norm sq. = 0.013508661650121212
elapsed = 0.040036916732788086s
Episode 39:
excess fwd cost = 35.674407958984375
bwd cost = 4.281457586330362e-05
bwd x0 - x0 = [-7.3909760e-06 4.4107437e-06]
fwd xT = [-0.08831382 0.0423443 ]
fwd xT norm sq. = 0.009592370130121708
elapsed = 0.03787875175476074s
Episode 40:
excess fwd cost = 34.967288970947266
bwd cost = 7.932206790428609e-05
bwd x0 - x0 = [-1.2516975e-05 6.0796738e-06]
fwd xT = [-0.07421579 0.03292185]
fwd xT norm sq. = 0.006591831799596548
elapsed = 0.0354008674621582s
Episode 41:
excess fwd cost = 34.314125061035156
bwd cost = 8.296375017380342e-05
bwd x0 - x0 = [-9.2983246e-06 4.7683716e-06]
fwd xT = [-0.06114713 0.02438434]
fwd xT norm sq. = 0.004333568271249533
elapsed = 0.033315181732177734s
Episode 42:
excess fwd cost = 33.70890426635742
bwd cost = 1.1069082574977074e-05
bwd x0 - x0 = [-5.2452087e-06 4.2915344e-06]
fwd xT = [-0.04900599 0.01662632]
fwd xT norm sq. = 0.0026780213229358196
elapsed = 0.029833078384399414s
Episode 43:
excess fwd cost = 33.146488189697266
bwd cost = 1.751014315232169e-05
bwd x0 - x0 = [-2.5033951e-06 1.1920929e-06]
fwd xT = [-0.03770227 0.00954989]
fwd xT norm sq. = 0.0015126613434404135
elapsed = 0.03327512741088867s
Episode 44:
excess fwd cost = 32.62240982055664
bwd cost = -3.3918084227479994e-05
bwd x0 - x0 = [ 5.9604645e-06 -2.9802322e-06]
fwd xT = [-0.02715517 0.00307947]
fwd xT norm sq. = 0.0007468863041140139
elapsed = 0.03175091743469238s
Episode 45:
excess fwd cost = 32.13275146484375
bwd cost = -3.265638224547729e-05
bwd x0 - x0 = [ 6.4373016e-06 -3.5166740e-06]
fwd xT = [-0.01729256 -0.00285523]
fwd xT norm sq. = 0.0003071848477702588
elapsed = 0.031952857971191406s
Episode 46:
excess fwd cost = 31.67401885986328
bwd cost = 9.116360888583586e-05
bwd x0 - x0 = [-1.3351440e-05 6.7949295e-06]
fwd xT = [-0.00804984 -0.00831582]
fwd xT norm sq. = 0.00013395267887972295
elapsed = 0.035826921463012695s
Episode 47:
excess fwd cost = 31.24310302734375
bwd cost = 2.1569801901932806e-05
bwd x0 - x0 = [ 9.059906e-06 -5.066395e-06]
fwd xT = [ 0.00063136 -0.01335132]
fwd xT norm sq. = 0.00017865627887658775
elapsed = 0.03789710998535156s
Episode 48:
excess fwd cost = 30.83730697631836
bwd cost = -7.845729123800993e-05
bwd x0 - x0 = [ 1.7166138e-05 -8.9406967e-06]
fwd xT = [ 0.00880232 -0.01800954]
fwd xT norm sq. = 0.00040182421798817813
elapsed = 0.027708053588867188s
Episode 49:
excess fwd cost = 30.45425796508789
bwd cost = -5.562108344747685e-05
bwd x0 - x0 = [ 1.04904175e-05 -5.54323196e-06]
fwd xT = [ 0.0165098 -0.02232964]
fwd xT norm sq. = 0.0007711865473538637
elapsed = 0.030406713485717773s
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment