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

Multi GPU training #88

Open
haoxintian opened this issue Dec 18, 2024 · 2 comments
Open

Multi GPU training #88

haoxintian opened this issue Dec 18, 2024 · 2 comments

Comments

@haoxintian
Copy link

May I ask if your training code supports multi GPU training? How should I conduct multi GPU training?

@felipecadar
Copy link
Collaborator

Hello @haoxintian. Thank you for your interest in our work!

We currently do not support multi GPU training. Please check the issue #30.

@haoxintian
Copy link
Author

Thank you so much for your prompt and helpful response! I truly appreciate your support.

I have a question regarding the training process: After training the model for 160,000 iterations, I have accumulated numerous weight files. However, I'm unsure which one is the optimal. Could you please clarify whether the most optimal weight file is automatically selected (should I just use the last one?), or do I need to manually evaluate the results from the terminal output after each iteration to decide which weight file to use?

I look forward to your reply!

Best regards

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