Model Constraints Factory
Model constraints factory.
get_model_constraints(model_config)
Get the model constraints for the given model configuration.
Parameters:
-
model_config(UNetConfig) –The model configuration.
Returns:
-
ModelConstraints–The model constraints for the given model configuration.
Raises:
-
ValueError–If the model type is not supported.