From 31c404822a38c2e0e07ad9d2bf33f0650ff620dd Mon Sep 17 00:00:00 2001 From: Aladdin Persson Date: Wed, 24 Mar 2021 22:03:12 +0100 Subject: [PATCH] small revisions to code examples --- ML/Pytorch/Basics/pytorch_rnn_gru_lstm.py | 1 + ML/Pytorch/Basics/pytorch_simple_CNN.py | 1 + ML/Pytorch/Basics/pytorch_simple_fullynet.py | 2 +- 3 files changed, 3 insertions(+), 1 deletion(-) diff --git a/ML/Pytorch/Basics/pytorch_rnn_gru_lstm.py b/ML/Pytorch/Basics/pytorch_rnn_gru_lstm.py index f12d855..489caeb 100644 --- a/ML/Pytorch/Basics/pytorch_rnn_gru_lstm.py +++ b/ML/Pytorch/Basics/pytorch_rnn_gru_lstm.py @@ -130,6 +130,7 @@ for epoch in range(num_epochs): # gradient descent update step/adam step optimizer.step() + # Check accuracy on training & test to see how good our model def check_accuracy(loader, model): num_correct = 0 diff --git a/ML/Pytorch/Basics/pytorch_simple_CNN.py b/ML/Pytorch/Basics/pytorch_simple_CNN.py index d18d948..45c94e1 100644 --- a/ML/Pytorch/Basics/pytorch_simple_CNN.py +++ b/ML/Pytorch/Basics/pytorch_simple_CNN.py @@ -94,6 +94,7 @@ for epoch in range(num_epochs): # gradient descent or adam step optimizer.step() + # Check accuracy on training & test to see how good our model def check_accuracy(loader, model): num_correct = 0 diff --git a/ML/Pytorch/Basics/pytorch_simple_fullynet.py b/ML/Pytorch/Basics/pytorch_simple_fullynet.py index fa0a9f4..4e08a9d 100644 --- a/ML/Pytorch/Basics/pytorch_simple_fullynet.py +++ b/ML/Pytorch/Basics/pytorch_simple_fullynet.py @@ -95,7 +95,7 @@ for epoch in range(num_epochs): optimizer.step() -# Check accuracy on training & test to see how good our model +# Check accuracy on training & test to see how good our model is def check_accuracy(loader, model): num_correct = 0 num_samples = 0