Skip to content

Commit 3f1cb23

Browse files
Madderchsasank
authored andcommitted
Fix epoch accuracy (avoiding integer division) (pytorch#227)
1 parent 2543fee commit 3f1cb23

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

beginner_source/transfer_learning_tutorial.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ def train_model(model, criterion, optimizer, scheduler, num_epochs=25):
187187
running_corrects += torch.sum(preds == labels.data)
188188

189189
epoch_loss = running_loss / dataset_sizes[phase]
190-
epoch_acc = running_corrects / dataset_sizes[phase]
190+
epoch_acc = running_corrects.double() / dataset_sizes[phase]
191191

192192
print('{} Loss: {:.4f} Acc: {:.4f}'.format(
193193
phase, epoch_loss, epoch_acc))

0 commit comments

Comments
 (0)