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

About a question of test_training_cifar.py #13

Closed
lastonephy opened this issue Jul 16, 2022 · 2 comments
Closed

About a question of test_training_cifar.py #13

lastonephy opened this issue Jul 16, 2022 · 2 comments

Comments

@lastonephy
Copy link

lastonephy commented Jul 16, 2022

In line 200 and 201 of test_training_cifar.py:
dataset = IMBALANCECIFAR100(data_dir, train=True, download=True, transform=train_trsfm, imb_type=imb_type, imb_factor=test_imb_factor, reverse=reverse)
train_dataset = IMBALANCECIFAR100(data_dir, train=True, download=True, transform= TwoCropsTransform(train_trsfm), imb_type=imb_type, imb_factor=test_imb_factor, reverse=reverse)
why you set the train is True? I think it should be False to obtain the weighting parameters of test set.
Can you explain it?
Thanks!

@Vanint
Copy link
Owner

Vanint commented Jul 16, 2022

Hi. thanks for mentioning this.
We double-check the test_training_cifar.py and find that this file is the wrong version when we clear up the open-source file.
In fact, you will obtain the results reported in our paper or on GitHub by changing Line 201 to

train_dataset = IMBALANCECIFAR100(data_dir, train=False, download=True, transform= TwoCropsTransform(train_trsfm), imb_type=imb_type, imb_factor=test_imb_factor, reverse=reverse)

We have modified the corresponding code file on Github. Thanks again.

@Vanint
Copy link
Owner

Vanint commented Jul 16, 2022

If you have any further questions, please let me know~

@Vanint Vanint closed this as completed Jul 16, 2022
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