lint:
black .
isort --profile=black .
flake8 --ignore E501 .
test:
python main.py --alpha 0.7 --lr 1e-3 -D 0.5 --max_epochs 1000
