GitHub repository: keskarnitish/large-batch-training
Primary language: Python
Description: Code to reproduce some of the figures in the paper "On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima"