Display pbar before starting training

odulcy-mindee · Mar 5, 2024 · 4ee8b4e · 4ee8b4e
1 parent 2f65f6e
commit 4ee8b4e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/references/detection/train_pytorch.py b/references/detection/train_pytorch.py
@@ -136,6 +136,7 @@ def fit_one_epoch(model, train_loader, batch_transforms, optimizer, scheduler, a
     last_progress = 0
     interval_progress = 5
     pbar = tqdm(train_loader, position=1)
+    send_on_slack(str(pbar))
     for images, targets in pbar:
         if torch.cuda.is_available():
             images = images.cuda()

diff --git a/references/detection/train_tensorflow.py b/references/detection/train_tensorflow.py
@@ -117,6 +117,7 @@ def fit_one_epoch(model, train_loader, batch_transforms, optimizer, amp=False):
     last_progress = 0
     interval_progress = 5
     pbar = tqdm(train_iter, position=1)
+    send_on_slack(str(pbar))
     for images, targets in pbar:
         images = batch_transforms(images)