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-04-16 15:13:46 +09:00
parent bd8394433f
commit 35586d6bca
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
resnet_finetune.py
View File

@ -95,7 +95,7 @@ for i in range(epochs):
val_acc = .0 val_acc = .0
model_gpu.train() model_gpu.train()
for images, labels in tqdm(dataloader['train'], leave=False): for images, labels in dataloader['train']:
optimizer.zero_grad() optimizer.zero_grad()
images = images.to(device) images = images.to(device)
labels = labels.to(device) labels = labels.to(device)