d1906629ab
Enable wit train on cutome dataset and loss down.
1ef3e419cb
Add custom dataset support.
e5f97af291
Add wit train support.