Dean
Training for 1 epoch, still need to fix evaluation step
ad95853