-
Notifications
You must be signed in to change notification settings - Fork 5
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
Collaborate #2
Comments
hi there! I am open to collaboration on interesting works. You may want to discuss your ideas and implementation details with me? best, |
Hi, I am relatively new to training diffusion models. I have only
fine-tuned ESM-2 models for sequence classification and for token
classification. Are you using `EsmForProteinFolding` as the backbone in
your diffusion model? If so, I don't believe I have access to a good enough
GPU to train it. My GPUs are too small unless a smaller model can be used.
I hope that I am wrong, or that another ESM-2 model can be used that is
smaller. Otherwise I am stuck and unable to train. I am having trouble
understanding your code also and was hoping we might work on writing a
notebook similar to this:
https://colab.research.google.com/github/huggingface/notebooks/blob/main/diffusers/training_example.ipynb
Thanks for responding!
Amelie
…On Thu, Aug 31, 2023 at 9:43 PM Zhangzhi Peng ***@***.***> wrote:
hi there! I am open to collaboration on interesting works. You may want to
discuss your ideas and implementation details with me?
best,
zhangzhi
—
Reply to this email directly, view it on GitHub
<#2 (comment)>,
or unsubscribe
<https://github.com/notifications/unsubscribe-auth/AMIK6IGP3CHAHK3NDFWIGATXYFRYBANCNFSM6AAAAAA4G2PBNE>
.
You are receiving this because you authored the thread.Message ID:
***@***.***
com>
|
hi, the training is pretty cheap. I can fit the model in a 10g GPU. Regarding the documentation, please follow the readme to install pkgs and train the model. Please let me know which parts confuse you. best, |
Could you find me on discord? EDIT: I've tried training on a P100 GPU (using a colab instance) and it doesn't seems to work. My training script must not be setup correctly or something. |
Hi,
|
Hi! I tried following the install instruction and I am having some issues. First, there seems to be a mistake in the install instructions. I believe you need
instead of
Also. Once everything is installed, I am getting the following error:
|
I guess the error is because the pytorch lightning version is updated and they stop using gpus as an argument. use trainer = pl.Trainer(max_epochs=20,accelerator="auto") |
I'm very interested in replicating your work and would like to train a diffusion model to generate protein binding partners similar to what RFDiffusion accomplishes, but I would like to use ESM-2 models as you have done. If you are open to collaborating, feel free to reach out if you have the time. Also, would you be able to create a tutorial similar to this?
The text was updated successfully, but these errors were encountered: