nh.modelzoo

neuralhydrology.modelzoo.get_model(cfg: Config) Module

Get model object, depending on the run configuration.

Parameters:

cfg (Config) – The run configuration.

Returns:

A new model instance of the type specified in the config.

Return type:

nn.Module