File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 22"""Train a network."""
33
44import torch
5- import warnings
65
76from ._workflow_utils import set_workflow_state
87from 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 ===
You can’t perform that action at this time.
0 commit comments