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

I want to train own datasets by this template. #66

Open
JasonChenhx opened this issue Feb 14, 2020 · 2 comments
Open

I want to train own datasets by this template. #66

JasonChenhx opened this issue Feb 14, 2020 · 2 comments

Comments

@JasonChenhx
Copy link

I want to train own datasets by this template. So How do I modify the code?

@EzoBear
Copy link

EzoBear commented Feb 18, 2020

you can use other dataset by modifying just some. follow details.

step is constructed of 2 phase.

  1. Make DataLoader.
  2. Modify Configuration File.

first step, you can see "dataloader class" from path that pytorch-template/data_loader/data_loaders.py.

Let's extends from base dataloader class.
in overrided init function, you make dataset that is supported pytorch class. if not supported dataset, you should make dataset class first. after call super's init function and put it in the first parameter.

second step, you modify config.json file from project's root directory. this template project read 'config.json' and load class file. so you just change data_loader's type from MnistDataLoader to your dataset class name created first step.

@JasonChenhx
Copy link
Author

JasonChenhx commented Feb 18, 2020 via email

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