yamle.utils.running_utils module#

yamle.utils.running_utils.extract_kwargs(args, prefix)[source]#

This method is used to extract the kwargs from the args.

Parameters:

  • args (argparse.Namespace) – The arguments.

  • prefix (str) – The prefix to be used for the arguments.

Return type:

Dict[str, Any]

Returns:

Dict[str, Any] – The extracted arguments.

yamle.utils.running_utils.prepare_datamodule_kwargs(args)[source]#

This method is used to prepare the datamodule kwargs.

Return type:

Dict[str, Any]

yamle.utils.running_utils.prepare_model_kwargs(args, datamodule)[source]#

This method is used to prepare the model kwargs.

Return type:

Dict[str, Any]

yamle.utils.running_utils.prepare_loss_kwargs(args, datamodule)[source]#

This method is used to prepare the loss kwargs.

Return type:

Dict[str, Any]

yamle.utils.running_utils.prepare_regularizer_kwargs(args)[source]#

This method is used to prepare the regularizer kwargs.

Return type:

Dict[str, Any]

yamle.utils.running_utils.prepare_metrics_kwargs(args, datamodule)[source]#

This method is used to prepare the metrics kwargs.

Return type:

Dict[str, Any]

yamle.utils.running_utils.prepare_method_kwargs(args, datamodule)[source]#

This method is used to prepare the method kwargs.

Return type:

Dict[str, Any]

yamle.utils.running_utils.prepare_trainer_kwargs(args, datamodule)[source]#

This method is used to prepare the trainer kwargs.

Return type:

Dict[str, Any]

yamle.utils.running_utils.prepare_test_trainer_kwargs(args, datamodule)[source]#

This method is used to prepare the trainer kwargs.

Return type:

Dict[str, Any]

yamle.utils.running_utils.prepare_pruner_kwargs(args)[source]#

This method is used to prepare the pruner kwargs.

Return type:

Dict[str, Any]

yamle.utils.running_utils.prepare_quantizer_kwargs(args)[source]#

This method is used to prepare the quantizer kwargs.

Return type:

Dict[str, Any]