python.object-model¶
model_attr_mutation¶
Original source code:
# mypy: allow-untyped-defs
import torch
class ModelAttrMutation(torch.nn.Module):
"""
Attribute mutation is not supported.
"""
def __init__(self):
super().__init__()
self.attr_list = [torch.randn(3, 2), torch.randn(3, 2)]
def recreate_list(self):
return [torch.zeros(3, 2), torch.zeros(3, 2)]
def forward(self, x):
self.attr_list = self.recreate_list()
return x.sum() + self.attr_list[0].sum()
Result:
AssertionError: Mutating module attribute attr_list during export.
optional_input¶
Original source code:
# mypy: allow-untyped-defs
import torch
class OptionalInput(torch.nn.Module):
"""
Tracing through optional input is not supported yet
"""
def forward(self, x, y=torch.randn(2, 3)):
if y is not None:
return x + y
return x
Result:
AssertionError: Unexpectedly found a <class 'torch.Tensor'> in the inputs.