nit: fixes

nikp1172 · nikp1172 · commit ea63c3e90c20 · 2024-02-09T18:18:25.000+05:30
diff --git a/mnist-classifaction/train_job/deploy.py b/mnist-classifaction/train_job/deploy.py
@@ -25,8 +25,8 @@
     resources=Resources(
        cpu_request=0.5,
        cpu_limit=0.5,
-       memory_request=1000,
-       memory_limit=1500
+       memory_request=1500,
+       memory_limit=2000
     )
     
 )
diff --git a/mnist-classifaction/train_job/train.py b/mnist-classifaction/train_job/train.py
@@ -42,7 +42,6 @@
     plt.axis('off')
 run.log_plots({"images": plt})
 plt.tight_layout()
-plt.show()
 
 
 # Define the model architecture
