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

update
continuous-integration/drone/push Build is passing Details

This commit is contained in:
yayoimizuha 2023-10-15 16:03:32 +09:00
parent 4b8480ffec
commit c6eab4b28b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
finetune/resnet_finetune.py
View File

@ -6,7 +6,7 @@ from torchvision.transforms import Compose, RandomResizedCrop, RandomRotation, T
Resize, CenterCrop, RandomAffine, GaussianBlur, RandomAutocontrast Resize, CenterCrop, RandomAffine, GaussianBlur, RandomAutocontrast
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from numpy import arange from numpy import arange
from torchsummary import summary from torchinfo import summary
from torch.nn import CrossEntropyLoss from torch.nn import CrossEntropyLoss
from torch.optim import SGD, Adam, lr_scheduler from torch.optim import SGD, Adam, lr_scheduler
from torchvision.datasets import ImageFolder from torchvision.datasets import ImageFolder