
export CUDA_VISIBLE_DEVICES=5 && python -u test_madelon.py --epochs 100

