[仅参数名不一致]torch.profiler.schedule

torch.profiler.schedule

torch.profiler.schedule(*, wait, warmup, active, repeat=0, skip_first=0)

paddle.profiler.make_scheduler

paddle.profiler.make_scheduler(*, closed, ready, record, repeat=0, skip_first=0)

两者功能一致,参数名不一致,具体如下:

参数映射

PyTorch PaddlePaddle 备注
wait closed 处于 wait/ProfilerState.CLOSED 状态的 step 数量,仅参数名不一致。
warmup ready 处于 warmup/ProfilerState.READY 状态的 step 数量,仅参数名不一致。
active record 处于 active/ProfilerState.RECORD 状态的 step 数量,仅参数名不一致。
repeat repeat 重复次数。
skip_first skip_first 首次 skip 步骤数量。