celldreamer/models/__init__.py

import yaml
from box import Box

def load_config(path):
    with open(path, 'r') as f:
        args = Box(yaml.safe_load(f))
        args.learning_rate = float(args.learning_rate)
        args.weight_decay = float(args.weight_decay)

    return args