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
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