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

About the Code #1

Open
AmingWu opened this issue Nov 28, 2023 · 7 comments
Open

About the Code #1

AmingWu opened this issue Nov 28, 2023 · 7 comments

Comments

@AmingWu
Copy link

AmingWu commented Nov 28, 2023

Dear Authors,

Have you released all the code?

@whcpumpkin
Copy link
Owner

Hi,
I am still refactoring the code. The current code does not include training and testing code. I will release all the code as soon as possible.

@AmingWu
Copy link
Author

AmingWu commented Dec 1, 2023

Dear Authors,

How long does your method need to train?

@whcpumpkin
Copy link
Owner

Dear Authors,

How long does your method need to train?

Hi,
On an A100 40G, each epoch takes about 4 hours, and approximately 20 epochs are needed to obtain the optimal model.
I think that increasing the number of workers in the dataloader can effectively reduce the train time, but this requires a sufficient number of CPU cores.

@AmingWu
Copy link
Author

AmingWu commented Dec 1, 2023

Does this code support multiple GPUs?

@whcpumpkin
Copy link
Owner

Does this code support multiple GPUs?

Currently, it only supports single GPU.

@AmingWu
Copy link
Author

AmingWu commented Dec 5, 2023

Dear Authors,

How to perform test?

@whcpumpkin
Copy link
Owner

Dear Authors,

How to perform test?

Hi,
I released the testing code just now.
If it has any problem, please let me know.

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