-
Notifications
You must be signed in to change notification settings - Fork 13
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
how to train my own dataset #2
Comments
In order to train on your dataset: Dataset Preparation
In order to merge both Imagenet and alov, I have used a custom
Each of the scripts is individually debuggable, please go through each of them in the above order to understand the functionality. Training Script
Let me know if you face any issues. |
@nrupatunga thk for your reply, I just find my machine can only support pytorch 1.1 at most( I can not upgrade the driver of the machine),when I use pytorch 1.4 which is in requirements.txt, then cuda can not work, can this repo support pytorch 1.1 ?if it can,where should I change? |
Though I haven't tested with PyTorch-1.1, It should work. You can refer https://pytorch.org/get-started/previous-versions/ for previous versions of PyTorch. |
In order to understand the current working of the data loaders, you can use a smaller dataset to understand it, which you can download it here |
@nrupatunga thanks for your quick reply!!.I have install pytorch 1.1 and cuda can work, and I test the repo, it always |
This might be the case, where you are missing other dependencies. |
ok, thank you very much!!! |
the imagenet dataset is to big, and I want to train with my own dataset , can this repo support train own dataset?very appreciate for your reply!
The text was updated successfully, but these errors were encountered: