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

update

This commit is contained in:
yayoimizuha 2023-05-04 22:53:25 +09:00
parent 7ca0bfa03c
commit a7290e57b4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
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 'layer3' in name: if 'layer1' in name:
tune = True tune = True
layer.requires_grad = tune layer.requires_grad = tune