towhee.trainer.scheduler.check_scheduler

towhee.trainer.scheduler.check_scheduler(scheduler_type: str) → bool

Check if the scheduler type is supported.

Parameters:

scheduler_type (str) – the type of the scheduler.

Return (bool):

if the scheduler type is supported.

Example

>>> from towhee.trainer.scheduler import check_scheduler
>>> check_scheduler('constant')
True