-
Notifications
You must be signed in to change notification settings - Fork 6
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
code for multi-gpu training #7
Comments
We didn't implement the multi-gpu training, since we just train a single linear layer, the training of LaBo is very fast (one epoch of entire ImageNet only takes around 10 minutes). The time-consuming part of the code could be the feature precompute, but it will only run once. Feel free to implement multi-gpu by yourself and I believe it is easy to do with pytorch-lightning. |
Many thanks for your help, I will try it later. File "/opt/conda/lib/python3.7/site-packages/apricot/functions/base.py", line 202, in fit when execute the code |
check this: #1 (comment) |
hi,it seems like the code provided is only for one GPU training, could you please provide the code for multi-gpu training?
The text was updated successfully, but these errors were encountered: