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)