added a bunch of verbose in train