diff --git a/references/detection/train_pytorch.py b/references/detection/train_pytorch.py index d23111f189..fccc092f1b 100644 --- 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 index 0b61c4c29e..62788331f0 100644 --- 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)