lint:
black .
isort --profile=black .
flake8 --ignore E501,W503 .
test:
python main.py --alpha 4 --lr 2e-4 --max_epochs 200
search:
python search.py