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.

Sign up for GitHub

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

Doesn't handle multi gpus #2

Open

qanastek opened this issue Aug 14, 2021 · 1 comment

Assignees

Labels

bug enhancement

Projects

Milestone

Owner

qanastek commented Aug 14, 2021

Please refer to https://pytorch-lightning.readthedocs.io/en/latest/advanced/multi_gpu.html

qanastek added the bug label

qanastek self-assigned this

qanastek added this to the Beta milestone

qanastek added this to To do in kanboard via automation

qanastek added the enhancement label

longswordinhand commented Jan 21, 2022

I soloved it using:
torch.multiprocessing.freeze_support()
and set
nbr_gpus=1 insider ObjectDetectionTrainer

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment