Cleanup
Cleanup
Cleanup
Continue cleanup
Cleanup
Change the type of data
Refactor prediction
Fix mnist reading
Cleanup pretrain