From 338614f6d905076ae7215c30084840e573847add Mon Sep 17 00:00:00 2001
From: Arnav Chavan <44640932+Arnav0400@users.noreply.github.com>
Date: Thu, 12 Aug 2021 12:04:12 +0530
Subject: [PATCH] removed save_weights

---
 runner.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/runner.py b/runner.py
index e8615e8..4eaa8d9 100644
--- a/runner.py
+++ b/runner.py
@@ -6,8 +6,8 @@
 import random
 
 # adapt these to you setup
-NR_GPUS = 2
-NR_PROCESSES = 2
+NR_GPUS = 4
+NR_PROCESSES = 4
 
 cnt = -1
 
@@ -23,7 +23,7 @@ def call_script(scripts):
     gpu = cnt % NR_GPUS
     crt_env['CUDA_VISIBLE_DEVICES'] = str(gpu)
     print(scripts)
-    sp.call([sys.executable, 'main.py', '--model_name', str(model), '--dataset', str(dataset), '--num_classes', str(num_classes), '--epochs', str(epochs), '--save_weights', '--weight_activation', wa], env=crt_env)
+    sp.call([sys.executable, 'main.py', '--model_name', str(model), '--dataset', str(dataset), '--num_classes', str(num_classes), '--epochs', str(epochs), '--weight_activation', wa], env=crt_env)
 
 
 if __name__ == '__main__':
@@ -40,4 +40,4 @@ def call_script(scripts):
     pool.map(call_script, scripts)
     pool.close()
     pool.join()
-    
\ No newline at end of file
+