From 10b3f04ca959c1ea9ee6979a1e3feeaa968854a2 Mon Sep 17 00:00:00 2001
From: mm <mm@clfx.cc>
Date: Sun, 31 Dec 2023 07:02:43 +0000
Subject: [PATCH] updates

---
 main.py  | 1 +
 makefile | 5 ++++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/main.py b/main.py
index a7d5fd5..1d59ddf 100644
--- a/main.py
+++ b/main.py
@@ -44,6 +44,7 @@ if __name__ == "__main__":
     args = parse_args()
 
     # Initialize data loader with parsed arguments
+    # named_data_loader also has grayscale extras. TODO: remove unnamed
     train_dataloader = init_data(
         batch_size=args.bs,
         shuffle=True,
diff --git a/makefile b/makefile
index ed2be99..950c38b 100644
--- a/makefile
+++ b/makefile
@@ -4,4 +4,7 @@ lint:
 	flake8 --ignore E501 .
 
 test:
-	python main.py --alpha 1 --lr 1e-3 --max_epochs 100
\ No newline at end of file
+	python main.py --alpha 1 --lr 1e-4 --max_epochs 500
+
+search:
+	python search.py