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

Update resnet_finetune.py
continuous-integration/drone/push Build is passing Details

This commit is contained in:
yayoimizuha 2023-04-17 18:00:23 +09:00 committed by GitHub
parent 35586d6bca
commit e4308ca509
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
resnet_finetune.py
View File

@ -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()