benchmarking.utils.get_cost_model module

benchmarking.utils.get_cost_model.get_cost_model_for_batch_size(params, batch_size_key, batch_size_range)[source]

Returns cost model depending on the batch size only.

Parameters:

  • params (Dict[str, Any]) – Command line arguments

  • batch_size_key (str) – Name of batch size entry in config

  • batch_size_range (Tuple[int, int]) – (lower, upper) for batch size, both sides are inclusive

Returns:

Cost model (or None if dependencies cannot be imported)