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

4
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
@@ -54,4 +54,4 @@ exp.fit_step(
with_tb=True, logdir='./finetuning_log', schedule_key='loss', num_workers=8,
pin_memory=False, follow_batch=[], exclude_keys=[], collate_fn=None,
max_edge_num_in_batch=400000
)
)