Skip to Content
Deep Learning with PyTorch
book

Deep Learning with PyTorch

by Vishnu Subramanian
February 2018
Intermediate to advanced
262 pages
6h 59m
English
Packt Publishing
Content preview from Deep Learning with PyTorch

Training and validating the model

We use the same fit and training logic as seen in the previous ResNet and other examples. We will just look at the training code and the results from it:

for epoch in range(1,10):    epoch_loss, epoch_accuracy = fit(epoch,fc,trn_feat_loader,phase='training')    val_epoch_loss , val_epoch_accuracy = fit(epoch,fc,val_feat_loader,phase='validation')    train_losses.append(epoch_loss)    train_accuracy.append(epoch_accuracy)    val_losses.append(val_epoch_loss)    val_accuracy.append(val_epoch_accuracy)#Resultstraining loss is 0.78 and training accuracy is 22825/23000 99.24validation loss is 5.3 and validation accuracy is 1947/2000 97.35training loss is 0.84 and training accuracy is 22829/23000 99.26validation loss is 5.1 and ...
Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.
Start your free trial

You might also like

Deep Learning with PyTorch

Deep Learning with PyTorch

Eli Stevens, Thomas Viehmann, Luca Pietro Giovanni Antiga
Grokking Deep Learning

Grokking Deep Learning

Andrew W. Trask

Publisher Resources

ISBN: 9781788624336Supplemental Content