update

2023-05-04 22:53:25 +09:00 · 2023-05-04 22:53:25 +09:00 · a7290e57b4
parent 7ca0bfa03c
commit a7290e57b4
1 changed files with 1 additions and 1 deletions
--- a/resnet_finetune_vggface.py
+++ b/resnet_finetune_vggface.py
@ -111,7 +111,7 @@ print()

 tune = False
 for name, layer in model.named_parameters():
-    if 'layer3' in name:
+    if 'layer1' in name:
        tune = True
    layer.requires_grad = tune