Update PythonApplication1.py's comment at line 48

PythonApplication1.py

@@ -45,7 +45,7 @@ def accuracy(outputs, labels):
 train_loader = DataLoader(train_ds, batch_size=batch_size, shuffle=True)
 val_loader = DataLoader(val_ds, batch_size=batch_size)
 
-# Creating a logistic regression model using nn.Linear
+# Declaring the size of the images and the no. of digits the number might fall into i.e. 0 to 9 is numclasses
 ip_size = 28 * 28
 num_classes = 10
 
@@ -223,4 +223,4 @@ def predict_image(img, model):
 # Evaluate the model after fine-tuning
 print("Evaluation after fine-tuning:")
 result_after_fine_tune = evaluate(model2, test_loader)
-print(result_after_fine_tune)
+print(result_after_fine_tune)