Initial commit- v1 trained on 1029 samples 1 epoch lr 1e-04

initial commit