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

Compare commits

base: tomokazu:730e910a36d7e77059da0eafe02f3ce86ab2bbfc
Branches Tags
tomokazu:master
tomokazu:2024-09-11
..
compare: tomokazu:7ca0bfa03c04f4a4baba6f7674403ede0b651375
Branches Tags
tomokazu:master
tomokazu:2024-09-11

No commits in common. "730e910a36d7e77059da0eafe02f3ce86ab2bbfc" and "7ca0bfa03c04f4a4baba6f7674403ede0b651375" have entirely different histories.
730e910a36 ... 7ca0bfa03c

2 changed files with 3 additions and 3 deletions
Show Stats 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 resnet_finetune_vggface.py - CI=True python resnet_finetune_vggface.py

4
resnet_finetune_vggface.py
View File

@ -111,7 +111,7 @@ print()
tune = False tune = False
for name, layer in model.named_parameters(): for name, layer in model.named_parameters():
if 'layer1' in name: if 'layer3' in name:
tune = True tune = True
layer.requires_grad = tune layer.requires_grad = tune
@ -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, mininterval=20)): for count, (images, labels) in enumerate(tqdm(dataloader['train'], disable=CI)):
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)