optuna.integration.ChainerMNStudy

class optuna.integration.ChainerMNStudy(study, comm)[source]

A wrapper of Study to incorporate Optuna with ChainerMN.

See also

ChainerMNStudy provides the same interface as Study. Please refer to optuna.study.Study for further details.

See the example if you want to optimize an objective function that trains neural network written with ChainerMN.

Parameters

Methods

optimize(func[, n_trials, timeout, catch])

Optimize an objective function.

optimize(func, n_trials=None, timeout=None, catch=())[source]

Optimize an objective function.

This method provides the same interface as optuna.study.Study.optimize() except the absence of n_jobs argument.

Parameters
  • func (Callable[[ChainerMNTrial, CommunicatorBase], float]) –

  • n_trials (Optional[int]) –

  • timeout (Optional[float]) –

  • catch (Tuple[Type[Exception], ..]) –

Return type

None