From 185646c601c56aed7bbfc5f48e93f40c2721073d Mon Sep 17 00:00:00 2001 From: yayoimizuha Date: Sun, 20 Aug 2023 00:42:36 +0900 Subject: [PATCH] update --- .drone.yml | 2 +- facenet_transfer_learning.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.drone.yml b/.drone.yml index 43e328a..dfb85fb 100644 --- a/.drone.yml +++ b/.drone.yml @@ -22,4 +22,4 @@ steps: - mkdir -p data - $mount_command - ls data/ - - CI=False python facenet_transfer_learning.py \ No newline at end of file + - CI=True python facenet_transfer_learning.py \ No newline at end of file diff --git a/facenet_transfer_learning.py b/facenet_transfer_learning.py index 34cfd29..5edc5b6 100644 --- a/facenet_transfer_learning.py +++ b/facenet_transfer_learning.py @@ -151,7 +151,7 @@ for epoch in range(epochs): model_gpu.train() # makedirs(join(save_dir, 'pallets', str(epoch)), exist_ok=True) - for count, (images, labels) in enumerate(tqdm(dataloader['train'])): + for count, (images, labels) in enumerate(tqdm(dataloader['train'], disable=None)): if count == 1: image_pallets = plot_dataset(dataloader=(images, labels), col_len=6, label_text=image_folder['train'].classes)