Kevin P Murphy murphyk
murphyk
/ gist:30b7ae562e392d437d3b089d926c7735
Created
May 25, 2021 04:08
uncertainty baselines table
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| add a (14x7) table of methods * tasks, where each cell lists the metrics ("at least 5") available for that combinaiton. |
murphyk
/ gist:c1871bfa8c40dc1a93258e52afa2c5d1
Last active
May 21, 2021 20:43
JL training pseudocode
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| model = jaxLightningModule() | |
| def fit(model): | |
| opt = model.configure_optimizers() | |
| opt_state = opt.init(model.params) | |
| for iter | |
| for batch | |
| losses = model.step(batch) | |
| grads = jax.grad(losses, model.params) | |
| grad = model.reduce_gradients(grads) | |
| param_update, opt_state = opt.update(grad, opt_state, model.params) |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| gist |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| See attached screenshot |
NewerOlder