Adds `experiment_config_json` for setting custom configurations with a json.

870e0db3 · Jennifer · 0ba7e7f9 · 870e0db3
Commit 870e0db3 authored Mar 13, 2024 by Jennifer
Hide whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

train_openfold.py train_openfold.py +0 -1

No files found.
--- a/train_openfold.py
+++ b/train_openfold.py
@@ -43,7 +43,6 @@ from scripts.zero_to_fp32 import (
    get_fp32_state_dict_from_zero_checkpoint,
    get_global_step_from_zero_checkpoint
 )
-from scripts.zero_to_fp32 import get_optim_files, parse_optim_states, get_model_state_file

 from openfold.utils.logger import PerformanceLoggingCallback