Giters

alinlab / M2m

Code for the paper "M2m: Imbalanced Classification via Major-to-minor Translation" (CVPR 2020)

Home Page:http://arxiv.org/abs/2004.00431

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

train M2m

wangjl1993 opened this issue · comments

commented

great work!
it seems to should add "--resume" when training M2m. (load ckpt_g) friendly warning.
by the way, do you apply M2m on larger dataset, such as Imagenet?

commented

I found the Imagenet experiment in supplementary material~ lol

commented

Hi, it's nice to hear that you found the experimental results of Imagenet!

Also, as M2m training requires pre-trained classifier g, "--resume" option should be added to apply it correctly.

But, regarding the warning message about "--resume", I can't find such a message in my environment when I turn off that option.

Can you tell me about it in more detail? Then, I'll update the codes more clearly. Thanks!

commented

hi, @ bbuing9. According to your reply and source code, I think the command provided in README for "Train another network f using M2m with the pre-trained g" should add --resume option

commented

Thanks for the nice suggestion, @lipingcoding! For clarification, I have updated both README and run.sh by adding --resume option.