Commit Graph

9 Commits

Author SHA1 Message Date
37c537ca31 fix lightning, prepare sweeps 2022-08-27 11:21:00 +02:00
d8169e01bc finish lightningification\n\nTraining still seems borked 2022-08-24 19:18:20 +02:00
63da24f429 change a bunch of stuff, add wip lightning implementation 2022-08-24 16:25:12 +02:00
Nils Koch
70e4bf6fe1 add wandb, make compatible with ctd data 2022-05-30 16:13:06 +02:00
Ibai
c432996dd5 Set flow_init to optional 2022-04-08 20:45:33 +09:00
Ibai
d199d62b8d Adjusted mesgh_grid for ONNX conversion 2022-04-08 18:10:20 +09:00
Ibai
f0a57bb444 Fix implementation issues 2022-04-08 17:48:29 +09:00
Ibai
3396d2fd06 Fix CUDA inference 2022-04-08 12:00:32 +09:00
Ibai
fc524edd76 Initial commit 2022-04-08 00:18:27 +09:00