[Install environment]

$ conda env create -f environment.yml
$ source activate diem
$ conda install pytorch torchvision cudatoolkit=10.0 -c pytorch


[Training]

$ python demo_amortized_clustering.py --num_proto 4 --num_ems 3 --num_heads 5 --out_type select_best2 --tau 1e-2 --num_steps 200000

