set_scheduler

dragon.autograph.set_scheduler(scheduler='SIMPLE')[source]

Set the scheduler for symbolic graph.

Parameters:
  • scheduler ({'SIMPLE', 'FUSION'}, optional) – The scheduler type.