hookeai.model_architectures.rnn_base_model.optimization.hydra_optimization.hydra_wrapper

hydra_wrapper(process, dataset_paths, device_type='cpu')[source]

Wrapper of Hydra hyperparameter optimization main function.

Parameters:

  • process (str) – Hyperparameter optimization process.

  • dataset_paths (dict) – Hyperparameter optimization process required data sets (key, str) file paths (item, str).

  • device_type ({'cpu', 'cuda'}, default='cpu') – Type of device on which torch.Tensor is allocated.