From 730e910a36d7e77059da0eafe02f3ce86ab2bbfc Mon Sep 17 00:00:00 2001 From: yayoimizuha Date: Thu, 4 May 2023 22:55:27 +0900 Subject: [PATCH] update --- .drone.yml | 2 +- resnet_finetune_vggface.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.drone.yml b/.drone.yml index b5ea4e4..43d2c20 100644 --- a/.drone.yml +++ b/.drone.yml @@ -22,4 +22,4 @@ steps: - mkdir -p data - $mount_command - ls data/ - - CI=True python resnet_finetune_vggface.py \ No newline at end of file + - CI=False python resnet_finetune_vggface.py \ No newline at end of file diff --git a/resnet_finetune_vggface.py b/resnet_finetune_vggface.py index a99cfbf..bf843e1 100644 --- a/resnet_finetune_vggface.py +++ b/resnet_finetune_vggface.py @@ -157,7 +157,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'], disable=CI)): + for count, (images, labels) in enumerate(tqdm(dataloader['train'], disable=CI, mininterval=20)): if count == 1: image_pallets = plot_dataset(dataloader=(images, labels), col_len=6, label_text=image_folder['train'].classes)