KeyError: 'ClusterNet6cTwoHead'

[PHOSPHENES8]

Hi, when I tried to run "cluster part“, the command I used is :
python -m code.scripts.cluster.cluster_greyscale_twohead --model_ind 685 --arch ClusterNet6cTwoHead --mode IID --dataset MNIST --dataset_root /scratch/local/ssd/xuji/MNIST --gt_k 10 --output_k_A 50 --output_k_B 10 --lamb_A 1.0 --lamb_B 1.0 --lr 0.0001 --num_epochs 3200 --batch_sz 700 --num_dataloaders 5 --num_sub_heads 5 --crop_orig --crop_other --tf1_crop centre_half --tf2_crop random --tf1_crop_sz 20 --tf2_crop_szs 16 20 24 --input_sz 24 --rot_val 25 --no_flip --head_B_epochs 2 > out/sh10_gpu3_m685.out,
However, the result suggests that there is a KeyError, can you help me with this problem, thanks a lot!

[xu-ji]

Did you download the full repository with all the files? If ClusterNet6cTwoHead is defined where it should be in the files, there should not be any key errors.