utils

Utility functions for models.

Functions

models.utils.load_model_config(args, buffer_size=None)

Loads the configuration file for the model.

Parameters:

• args (Namespace) – the arguments which contains the hyperparameters

• buffer_size (int | None) – if a method has a buffer, knowing the buffer_size is required to load the best configuration

Returns:

the configuration of the model

Return type:

dict