Testing on other categories

[kakou34]

Hello, I am trying to reproduce the results obtained on the classes with small numbers of samples like mug and bottle but the evaluation step is throwing this error

AssertionError: file not found: ./datasets/test_data/ref_val_mug.pt

In the drive folder, you only provided the files for chair, car and airplane. Could you please assist me with how to generate these files for other categories?
Thanks !

[ZENGXH]

I upload mug and bottle here: bottle mug

When you call the compute_score function, please also turn the norm_box=True since the reference data is normalized to [-1,1] already.