updates

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,

makefile

@@ -4,4 +4,7 @@ lint:
 	flake8 --ignore E501 .
 
 test:
-	python main.py --alpha 1 --lr 1e-3 --max_epochs 100
+	python main.py --alpha 1 --lr 1e-4 --max_epochs 500
+
+search:
+	python search.py