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 is passing Details

This commit is contained in:
yayoimizuha 2023-08-20 00:42:36 +09:00
parent 2ffa97ba64
commit 185646c601
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=False python facenet_transfer_learning.py - CI=True python facenet_transfer_learning.py

2
facenet_transfer_learning.py
View File

@ -151,7 +151,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'])): for count, (images, labels) in enumerate(tqdm(dataloader['train'], disable=None)):
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)