From e4308ca5090bd96fe6d044d78649d393c2ba10c1 Mon Sep 17 00:00:00 2001
From: yayoimizuha <poko.peko.tomo@icloud.com>
Date: Mon, 17 Apr 2023 18:00:23 +0900
Subject: [PATCH] Update resnet_finetune.py

---
 resnet_finetune.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/resnet_finetune.py b/resnet_finetune.py
index 845e0cd..8437306 100644
--- a/resnet_finetune.py
+++ b/resnet_finetune.py
@@ -56,10 +56,10 @@ for name, layer in model.named_parameters():
         tune = True
     layer.requires_grad = tune
 
-print(model)
+# print(model)
 
 model.fc = Linear(in_features=2048, out_features=image_folder['train'].classes.__len__(), bias=True)
-summary(model=model, input_size=(3, 224, 224), device='cpu')
+# summary(model=model, input_size=(3, 224, 224), device='cpu')
 
 model_gpu = model.to(device=device)
 criterion = CrossEntropyLoss()