PipelineScheduleSingle

class tensorplay.distributed.pipelining.PipelineScheduleSingle(stage: Any, n_microbatches: int, loss_fn: Any = None, args_chunk_spec: Any = None, kwargs_chunk_spec: Any = None, output_merge_spec: Any = None, scale_grads: bool = True)[source]