Todo

add feature:

tensorboard for visualization
learning rate scheduler
dropout and weight decay