conda create -n FMI python=3.9
conda activate FMI
pip install -r requirements.txt

cd path/to/this/repo/
export PYTHONPATH=path/to/this/repo
python -m domainbed.scripts.sweep launch\
       --data_dir=/my/datasets/path\
       --output_dir=/my/sweep/output/path\
       --command_launcher MyLauncher\
       --algorithms FMI --datasets ColoredMNIST\ 
       --n_hparams 1 --n_trial 10

python -m domainbed.scripts.collect_results\
       --input_dir=/my/sweep/output/path

cd path/to/this/repo
sh scripts_FMI/unit_tests/sweep.sh

cd path/to/this/repo/
python scripts_unit_tests/sweep.py --num_iterations 10000 --num_data_seeds 50\ 
    --num_model_seed 20 --output_dir results/ --num_samples 1000

python scripts_unit_tests/collect_results.py results/COMMIT/

cd path/to/this/repo
python -m domainbed.attention.plot_attention --dataset ColoredMNIST\ 
    --data_dir domainbed/data/MNIST --model FMI --test_envs 2\ 
    --model_path scripts_FMI/model/model_ColoredMNIST_FMI.pkl --row 2 --col 2