From 686e096b975b4521ab0a42bc4030e315746972bf Mon Sep 17 00:00:00 2001 From: Michael Pilosov Date: Mon, 15 Jan 2024 19:26:59 +0000 Subject: [PATCH] saturate T4 GPU --- makefile | 2 +- search.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/makefile b/makefile index 8a1eb51..ba5e28b 100644 --- a/makefile +++ b/makefile @@ -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 diff --git a/search.py b/search.py index 802129d..c44d13b 100644 --- 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))