Можно ли загрузить конфиги из файла для «динамической группы конфигураций» в Hydra? ⇐ Python

[Anonymous]

Представьте себе сложную конфигурацию, в которой указано множество моделей машинного обучения, каждая из которых имеет определенное количество слоев. Что-то вроде

Код: Выделить всё

@dataclass
class Layer:
width: int
activation: str
...

@dataclass
class Model:
layers: List[Layer]

@dataclass
class Forest:
models: Dict[str, Model]

Затем у меня есть файл yaml в качестве основной конфигурации, в котором я определяю лес моделей. Затем в командной строке я могу переопределить отдельные поля некоторого слоя чем-то вроде models.cnn.layers.1.width=10. Я хотел бы иметь каталог с различными слоями и иметь возможность делать что-то вроде models.cnn.layers.1=wide_cnn в командной строке, где Wide_cnn.yaml — это какой-то файл yaml в этом каталоге.
Я пробовал разные вещи, например, использование Wide_cnn.yaml, использование абсолютного пути, использование models/cnn/layers.1 и т. д. Кажется, это основная проблема. заключается в том, что для того, чтобы гидра загружала файл, текст перед = в командной строке (назовем его «ключом») должен быть группой конфигурации. Я не совсем понимаю, что делает что-то группой конфигурации, но похоже, что это соответствует структуре каталогов. В моем случае ключом может быть что угодно, потому что пользователи могут называть свои модели произвольно.
Есть предложения, как сделать что-то подобное? Спасибо!

Подробнее здесь: https://stackoverflow.com/questions/790 ... p-in-hydra