message: training: train: print total duration, minor cleanup
diff: @@ -259,6 +259,11 @@ def main_worker(rank, cfg):
  # Save a checkpoint
  save_checkpoint(result_dir, epoch, step, model, optimizer)
  + # Print final stats
  + if rank == 0:
  + total_duration = time.time() - total_start_time
  + print('\nFinished (%s)' % format_time(total_duration))
  +
  # Cleanup
  cleanup_worker(cfg)
  