the script of training - Githubissues

BioinfoMachineLearning / CDPred

Deep transformer for predicting interchain residue-residue distances of protein complexes

MIT License

12 stars 3 forks source link

the script of training #10

Open onlyonewater opened 2 months ago

onlyonewater commented 2 months ago

can the authors provide the script or command about training from scratch? and How much GPU memory does CDPred require?

ZhiYeG commented 2 months ago

Please find the script ./lib/Model_training.py. And we training CDPred on one V100 with 32GB memory

onlyonewater commented 2 months ago

thanks for your reply, I found that the file in ./lib/train_homopred_tune_net.py seems to be the main function of training, and I want to know what these parameters mean. Can you give a simple example of using these parameters to train the network?