Commit Graph

4 Commits

Author SHA1 Message Date
Jake VanderPlas 0e5237df2a Use jax.api.device_put_sharded() in place of private JAX APIs.
PiperOrigin-RevId: 332514384
2020-09-23 16:43:36 +01:00
Florent Altché 8457046b2c Add checkpoints from the ablation study.
PiperOrigin-RevId: 328023346
2020-08-26 16:54:56 +01:00
Florent Altché 63fa5e72d5 Fix initial convolution channels not multiplied by width_multiplier.
PiperOrigin-RevId: 327005191
2020-08-21 08:21:04 +00:00
Clara Huiyi Hu 923ad3cff0 This notebook illustrates the CIFAR-10 experiments in the paper:
PiperOrigin-RevId: 326141025
2020-08-21 08:16:49 +00:00