Skip to content

GitLab

"lib/llm/src/git@developer.sourcefind.cn:OpenDAS/dynamo.git" did not exist on "9162f3ad2a9578072eac3c42bca7420041a5a6f1"
Commit a9cf892f authored by Geoffrey Yu's avatar Geoffrey Yu
Browse files

update feature_pipeline accordingly

parent 9aebc203
Hide whitespace changes
Inline Side-by-side
Showing with 13 additions and 5 deletions
openfold/data/feature_pipeline.py
View file @ a9cf892f
......@@ -93,14 +93,22 @@ def np_example_to_features(
with torch.no_grad():
if is_multimer:
features,gt_features = input_pipeline_multimer.process_tensors_from_config(
tensor_dict,
cfg.common,
cfg[mode],
)
if mode == 'train':
features,gt_features = input_pipeline_multimer.process_tensors_from_config(
tensor_dict,
cfg.common,
cfg[mode],
is_training=True
)
return {k: v for k, v in features.items()}, gt_features
else:
features = input_pipeline_multimer.process_tensors_from_config(
tensor_dict,
cfg.common,
cfg[mode],
is_training=False
)
return {k: v for k, v in features.items()}
else:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment