pub_soft/PocketFlow
3
0
Fork 0
mirror of https://github.com/Saoge123/PocketFlow.git synced 2026-06-04 12:44:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update finetuning.py

Browse Source
This commit is contained in:
Saoge123
2024-03-26 17:32:51 +08:00
committed by GitHub
parent 5a0bd0a8af
commit a6e9ca11cf
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
finetuning.py
View File

@@ -1,5 +1,5 @@
import torch
from pocket_flow.gdbp_model import PocketFlowWithEdgeNew, reset_parameters, freeze_parameters
from pocket_flow.gdbp_model import PocketFlow, reset_parameters, freeze_parameters
from pocket_flow.utils import Experiment, LoadDataset
from pocket_flow.utils.transform import *
#from utils.ParseFile import Protein, parse_sdf_to_dict