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

fine tuning peptide-MHC full model #8

Open
croshong opened this issue Jul 26, 2023 · 7 comments
Open

fine tuning peptide-MHC full model #8

croshong opened this issue Jul 26, 2023 · 7 comments

Comments

@croshong
Copy link

Hi

I'm now trying to reproduce the fine tuning model for peptide MHC model

Ithe example in readme shows the following script command line

for fine tuning peptide-MHC full model

python run_finetuning.py
--data_dir $ALPHAFOLD_DATA_DIR
--binder_intercepts 0.80367635 --binder_intercepts 0.43373787
--freeze_binder
--train_dataset datasets_alphafold_finetune/pmhc_finetune/combo_1and2_train.tsv
--valid_dataset datasets_alphafold_finetune/pmhc_finetune/combo_1and2_valid.tsv

I'm wondering whether the above example is just the example which shows the script usage or I can use the

model from above script for real production purpose

It the above command line is just example, what kind of parameter should I change for real production purpose ?

Thanks

@phbradley
Copy link
Owner

phbradley commented Jul 26, 2023 via email

@croshong
Copy link
Author

Thanks for your reply

I have successfully setup your code and alphafold library, several other library from deepmind

I'm now running above finetunning command, but it seems to take really very long time, more than several days

I started running 5 days ago, but it's till runnning in my server

My server has a GTX1080 GPU card which has a gpu computing capability around 6

what kind of gpu and server you are using for finetuing?

and Can you share the finetuning model generated with above command

which is used for PNAS publication?

Thanks

@phbradley
Copy link
Owner

phbradley commented Jul 30, 2023 via email

@croshong
Copy link
Author

attached files are log file for finetuning which is still currently running. it's still in training epoch 0

and I could find the finetuned parameter as you mentioned. but for the model_2_ptm, should I find it in the original alphafold package ?

@croshong
Copy link
Author

alphafold_log
I forgot the attachment

@croshong
Copy link
Author

I think model_2_ptm means the file in params/parmas_model2_2_ptm.npz
then what is the model_2_ptm_ft ? maybe should it be produced by finetuning?

@croshong
Copy link
Author

croshong commented Aug 1, 2023

I could run the prediction with finetuning paramter with following command line
python run_prediction.py --targets examples/pmhc_hcv_polg_10mers/targets.tsv
--outfile_prefix polg_test2 --model_names model_2_ptm_ft
--model_params_files datasets_alphafold_finetune/params/mixed_mhc_pae_run6_af_mhc_params_20640.pkl
--ignore_identities

and I got the
polg_test1_final.tsv.xlsx
table like attached excel file
with this table how can I get the binding score for each peptide and MHC?

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