The implication of "NUM_PROTOTYPE" in train_sim.py?

[YBIO]

Hi, thanks for your amazing work! I would like to know the function of the parameter "NUM_PROTOTYPE" in train_sim.py and train_sim_ssl.py. Is it relevant to the class number of the dataset? In addition, I tried to train the model on my own datasets (with only two classes. And I have added the corresponding scripts in "./dataset/" folder to load my own datasets), however, the performance is very poor, Do you have any suggestions? Thanks!