added to logging logic during training to be consistent with validation

Author: Tony

train.py

@@ -1264,7 +1264,7 @@ def _backward(_loss):
         update_time_m.update(time.time() - update_start_time)
         update_start_time = time_now
 
-        if update_idx % args.log_interval == 0:
+        if update_idx % args.log_interval == 0 or last_batch:
             lrl = [param_group['lr'] for param_group in optimizer.param_groups]
             lr = sum(lrl) / len(lrl)