Commit History
-
minor changes
db39442 -
feat(model, train): improved architecture, overfit prevention, re-trained the model
fa13b6c
-
feat(data): implemented data augmentation
5f436de
-
feat(train, dist_train, plots): improved training, implemented distributed training, added plotting
b42a2f2