diff --git a/main/_modules/torchtune/config/_utils.html b/main/_modules/torchtune/config/_utils.html index 46af67abbf..5b24e9c7bd 100644 --- a/main/_modules/torchtune/config/_utils.html +++ b/main/_modules/torchtune/config/_utils.html @@ -602,6 +602,11 @@

Source code for torchtune.config._utils

         # key string to reflect this
         if k in yaml_kwargs and _has_component(yaml_kwargs[k]):
             k += "._component_"
+
+        # None passed via CLI will be parsed as string, but we really want OmegaConf null
+        if v == "None":
+            v = "!!null"
+
         # TODO: this is a hack but otherwise we can't pass strings with leading zeroes
         # to define the checkpoint file format. We manually override OmegaConf behavior
         # by prepending the value with !!str to force a string type