From c6eab4b28b27bcdc9dae8a04a5881e3604022f8b Mon Sep 17 00:00:00 2001 From: yayoimizuha Date: Sun, 15 Oct 2023 16:03:32 +0900 Subject: [PATCH] update --- finetune/resnet_finetune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/finetune/resnet_finetune.py b/finetune/resnet_finetune.py index 5b3a74d..072ff32 100644 --- a/finetune/resnet_finetune.py +++ b/finetune/resnet_finetune.py @@ -6,7 +6,7 @@ from torchvision.transforms import Compose, RandomResizedCrop, RandomRotation, T Resize, CenterCrop, RandomAffine, GaussianBlur, RandomAutocontrast import matplotlib.pyplot as plt from numpy import arange -from torchsummary import summary +from torchinfo import summary from torch.nn import CrossEntropyLoss from torch.optim import SGD, Adam, lr_scheduler from torchvision.datasets import ImageFolder