From a7290e57b4fca2d5cc344f1d9d1427d9ae16ed70 Mon Sep 17 00:00:00 2001
From: yayoimizuha <poko.peko.tomo@icloud.com>
Date: Thu, 4 May 2023 22:53:25 +0900
Subject: [PATCH] update

---
 resnet_finetune_vggface.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/resnet_finetune_vggface.py b/resnet_finetune_vggface.py
index a873b1a..a99cfbf 100644
--- 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