Sebastian Caldas

https://github.com/TalwalkarLab/leaf/blob/b71d2716f27c613373c93eb0d5373f25063abe3d/models/main.py#L126 Names should be 'batch-size' and 'num-epochs'.