catenets.models.torch.utils.decorators module

benchmark(func: Callable) → Callable

Decorator used for function duration benchmarking. It is active only with DEBUG loglevel.

Parameters

func – the function to be benchmarked.

Returns

the decorator

Return type

Callable

check_input_train(func: Callable) → Callable

Decorator used for checking training params.

Parameters

func – the function to be benchmarked.

Returns

the decorator

Return type

Callable