Skip to content

GitLab

".github/git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "a5cf2a5362e752beb927bdf3f0785475cf24a64f"
Commit d4ed0b0c authored by Jennifer's avatar Jennifer
Browse files

Add log statement to weight conversion script

parent 67454c39
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
openfold/utils/import_weights.py
View file @ d4ed0b0c
...@@ -670,7 +670,6 @@ def import_jax_weights_(model, npz_path, version="model_1"): ...@@ -670,7 +670,6 @@ def import_jax_weights_(model, npz_path, version="model_1"):
def convert_deprecated_v1_keys(state_dict): def convert_deprecated_v1_keys(state_dict):
"""Update older OpenFold model weight names to match the current model code.""" """Update older OpenFold model weight names to match the current model code."""
logging.warning('converting keys...')
replacements = { replacements = {
'template_angle_embedder': 'template_single_embedder', 'template_angle_embedder': 'template_single_embedder',
......
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