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

training on Imagenet data is not working #75

Open
virtuoso108 opened this issue Oct 11, 2024 · 0 comments
Open

training on Imagenet data is not working #75

virtuoso108 opened this issue Oct 11, 2024 · 0 comments

Comments

@virtuoso108
Copy link

I am using one GPU so I ran the command like following.

python -m torch.distributed.launch --nproc_per_node=1 --master_port 12346 --use_env main.py --model repvit_m0_9 --data-path /media/iipl/DATA/DAT-main/imagenet --dist-eval

I trained for 300 epochs and the test acc1 is 0.094 and test acc5 is 1.338 which is very poor performance.
How can I train for classification on Imagenet with only one GPU?
Is there any learning rate and batch size for training with only one GPU?

I'd appreciate any help.
Thank you,

repvit_train_not_Working_results_screenshot

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