Right now we have a working system but it could stand some optimization. - [ ] Add ability to pick what optimizer we want to use. - [ ] More histogram data on torchboard.