search

LTH14 / targeted-supcon

A PyTorch implementation of the paper Targeted Supervised Contrastive Learning for Long-tailed Recognition
MIT License
93 stars 13 forks source link

class-target assignment code #4

Closed shuaiNJU closed 1 year ago

shuaiNJU commented 2 years ago

Hi, thanks for your share, and which one is the class-target assignment code?

twowwj commented 2 years ago

Did you find it?

LTH14 commented 1 year ago

The class-target assignment part is in moco/builder.py, from L354-L380.