added comments and more explicit training loop

upload model

initial commit