tomokazu
/
helloproject-ai
mirror of https://github.com/yayoimizuha/helloproject-ai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

update
continuous-integration/drone/push Build was killed Details

This commit is contained in:
yayoimizuha 2023-05-04 22:55:27 +09:00
parent a7290e57b4
commit 730e910a36
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.drone.yml
View File

@ -22,4 +22,4 @@ steps:
- mkdir -p data - mkdir -p data
- $mount_command - $mount_command
- ls data/ - ls data/
- CI=True python resnet_finetune_vggface.py - CI=False python resnet_finetune_vggface.py

2
resnet_finetune_vggface.py
View File

@ -157,7 +157,7 @@ for epoch in range(epochs):
model_gpu.train() model_gpu.train()
makedirs(join(save_dir, 'pallets', str(epoch)), exist_ok=True) 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: if count == 1:
image_pallets = plot_dataset(dataloader=(images, labels), col_len=6, image_pallets = plot_dataset(dataloader=(images, labels), col_len=6,
label_text=image_folder['train'].classes) label_text=image_folder['train'].classes)