Update train.py

added '(per batch)' in the print statement in the logs for the training process

Author: n0obcoder

train.py

@@ -148,7 +148,7 @@ def plot_losses(running_train_loss, running_val_loss, train_epoch_loss, val_epoc
         if (batch_idx + 1)%log_interval == 0:
             batch_time = time.time() - batch_start_time
             m,s = divmod(batch_time, 60)
-            print('train loss @batch_idx {}/{}: {} in {} mins {} secs'.format(str(batch_idx+1).zfill(len(str(len(train_loader)))), len(train_loader), loss.item(), int(m), round(s, 2)))
+            print('train loss @batch_idx {}/{}: {} in {} mins {} secs (per batch)'.format(str(batch_idx+1).zfill(len(str(len(train_loader)))), len(train_loader), loss.item(), int(m), round(s, 2)))
 
     train_epoch_loss.append(np.array(running_train_loss).mean())