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 @@
# 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