nit: fixes

truefoundry · Feb 9, 2024 · ea63c3e · ea63c3e
1 parent 4632942
commit ea63c3e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
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