how get os.cpu_count() in a yaml file?

[jaffe-fly]

i want to realize like this

trainer:
    cpu_count: os.cpu_count() 

how to do ?

[Jasha10]

Hi @jaffe-fly, I'd recommend using an OmegaConf custom resolver.

import os
from omegaconf import OmegaConf

OmegaConf.register_new_resolver("os_cpu_count", lambda: os.cpu_count())

my_yaml = """
trainer:
    cpu_count: ${os_cpu_count:}
"""

cfg = OmegaConf.create(my_yaml)
expected = os.cpu_count()
assert cfg.trainer.cpu_count == expected

[jaffe-fly]

thank you very much