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