Add unit tests to check for robustness to failures (i.e., checkpointing) for the JFT training scripts.
dusenberrymw opened this issue · comments
This was added in #490
google / uncertainty-baselines
High-quality implementations of standard and SOTA methods on a variety of tasks.