SDS-Lab / ROT-Pooling

Learnable Global Pooling Layers Based on Regularized Optimal Transport (ROT)
MIT License
15 stars 2 forks source link

ModuleNotFoundError: No module named 'img_classifier_learning_k' #1

Open fan23j opened 10 months ago

fan23j commented 10 months ago

Hello,

I can't seem to locate what img_classifier_learning_k is referring to in terms of an internal or external module.

in line 8 of resnet_imagenet.py: import img_classifier_learning_k as icl

minjiecheng commented 6 months ago

I'm sorry for the delayed response. I've updated the functions from img_classifier_learning_k to resnet_imagenet.py. You can try running it again.