saturate T4 GPU

10 months ago · 686e096b97
2 changed files with 2 additions and 2 deletions
--- a/2
+++ b/2
@ -4,7 +4,7 @@ lint:
 	flake8 --ignore E501,W503 .
 test:
-	python main.py --alpha 2 --lr 2e-4 --max_epochs 200
+	python main.py --alpha 2 --lr 2e-4 --max_epochs 200 --bs 8192
 search:
 	python search.py
--- a/search.py
+++ b/search.py
@ -23,7 +23,7 @@ learning_rate_values = list(np.round(np.logspace(-5, -3, 21), 5))
 alpha_values = [0, 1, 2]
 widths = [64, 128, 256, 512]
 # learning_rate_values = [5e-4]
-batch_size_values = [32, 64, 128]
+batch_size_values = [8192]
 max_epochs_values = [50]
 seeds = list(range(21, 1992))