Teli Ma

@wudizuixiaosa You just need to write your own Dataset Class in [datasets.py](https://github.com/Alpha-VL/ConvMAE/blob/main/util/datasets.py). Also, don't forget change the data_path and nb_classes in args of [main_finetune.py](https://github.com/Alpha-VL/ConvMAE/blob/main/main_finetune.py).

As the title, how to plot ROC curve to get the AUC on each test set? It would be great if you would give some sample code :)

And if I use your setup.py, it shows as : gcc: error: mmdet/ops/nms/src/rnms_ext.cpp: No such file or directory