torch.fx.experimental.debug.set_trace

torch.fx.experimental.debug.set_trace(gm) [исходный код]

Устанавливает точку останова в сгенерированном коде Python gm. При запуске gm выполнение перейдёт в pdb.

Параметры:

gm (GraphModule) – модуль графа, в который нужно вставить точку останова. Затем он перекомпилируется, чтобы изменения вступили в силу.

Возвращает:

gm со вставленной точкой останова.

Тип возвращаемого значения:

GraphModule