Skip to content

Commit fc3e34c

Browse files
committed
remove global_options TF32 migration warning from nequip-train
1 parent 705d3d0 commit fc3e34c

1 file changed

Lines changed: 1 addition & 17 deletions

File tree

nequip/scripts/train.py

Lines changed: 1 addition & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
"""Train a network."""
33

44
import torch
5-
import warnings
65

76
from ._workflow_utils import set_workflow_state
87
from nequip.utils import get_current_code_versions, RankedLogger
@@ -92,23 +91,8 @@ def main(config: DictConfig) -> None:
9291
f"and use the output directory provided by Hydra: {hydra.core.hydra_config.HydraConfig.get().runtime.output_dir}"
9392
)
9493

95-
logger.debug("Setting global options ...")
96-
97-
# as of NequIP v0.14.0, we removed the need to provide `global_options` in the config file
98-
# we provide this warning to help users migrate
99-
if "global_options" in config:
100-
warnings.warn(
101-
"\n\n!!! WARNING !!!\n"
102-
"The `global_options` section is no longer a required section in config files and will be ignored. "
103-
"TF32 settings should now be configured using the TF32Scheduler callback:\n\n"
104-
"callbacks:\n"
105-
" - _target_: nequip.train.callbacks.TF32Scheduler\n"
106-
" schedule:\n"
107-
" 0: true\n\n"
108-
"See the documentation for more details.\n",
109-
)
110-
11194
# === initialize global state ===
95+
logger.debug("Setting global options ...")
11296
set_global_state()
11397

11498
# === instantiate datamodule ===

0 commit comments

Comments
 (0)