Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Train/Test split when training linear classifier (ImageNet) #53

Open
jnyjxn opened this issue May 25, 2020 · 3 comments
Open

Train/Test split when training linear classifier (ImageNet) #53

jnyjxn opened this issue May 25, 2020 · 3 comments

Comments

@jnyjxn
Copy link

jnyjxn commented May 25, 2020

Hello,

Thank you for this great work and well-written paper.

When training the linear classifier (e.g. Alexnet with ImageNet), how do you perform the train/test split? 1. Do you use all of the classes in your test set?
2. How do you choose views for each sample?

Many thanks,
Jonny

@HobbitLong
Copy link
Owner

Hi Jonny,

  1. ImageNet dataset itself has a "train" set and a "val" set, which will be used as train and test sets, respectively.
  2. The same as the CMC pre-training stage. If you train CMC with Y/DbDr split, then you use Y/DbDr split in linear evaluation stage as well.

@jnyjxn
Copy link
Author

jnyjxn commented Jun 1, 2020

Thanks @HobbitLong - unfortunately from what I can tell ImageNet is defunct (I've had no response for > 3 months) so I can't access the dataset.

Could you briefly describe the approach they've used to do the split? Specifically, do they split according to e.g. 80 different chairs in train, 20 different chairs in val, or is it the same 100 chairs in train and val but with different perspectives?

Thank you!

@HobbitLong
Copy link
Owner

I believe it's the former way.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants