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

N-shot Training #38

Open
kerem0comert opened this issue May 19, 2023 · 1 comment
Open

N-shot Training #38

kerem0comert opened this issue May 19, 2023 · 1 comment

Comments

@kerem0comert
Copy link

Hello,
I would very simply like to train a model with the subsample you provide. I could get it to run by following the README.
At the root of the project I have:
ann_subsample/
subsample/
which I simply downloaded from the README and extracted. From the root of the project, I am running:

python -m classifier.run --command train --path_to_config classifier/config/default.yaml

At this point, I am able to train.

However, I would also like to train n-shot models, where I take n<5 samples per class and train with it. Then I would like to test this model. What would be the procedure for this? Because from what I see, no parameter to set train_test_percentage is provided in default.yaml. Ideally I would like to set train percentage at 1 in training (where the dataset is a subsample of let's say 3 images per gesture class) and then test it on another dataset (perhaps the test that you already provide)
How can I achieve this? Thanks!

@kerem0comert
Copy link
Author

Rather, I can ask the following:
I see that when training is complete, a pth model is saved under experiments/{model.name}/best_model.pth (I guess one is also saved each epoch?)
How to configure the params as such this pth model is used in the test?

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

1 participant