GB0246

Graph-Break Type

Short name describing what triggered the graph break

Attempted to use torch.nn.functional.one_hot with data-dependent output shape

Context

Values or code snippet captured at the break point

args={args}, kwargs={kwargs}

Explanation

Explanation of why the graph break was triggered

Dynamo does not support this.

Hints

Hints on how to resolve the graph break

• Explicitly set the num_classes param of the function call
• torch.nn.functional.one_hot to something other than -1.

Additional Information

Back to Registry