使用@trace装饰train_func时的参数问题

[NJUJet]

我的代码是在一个类中创建了self.model以及self.optimizer，按照如下方式调用trace装饰的train_func：
train_func(data, label, opt=self.optimizer, net=self.model)，报错信息如下：

如果以global model和global optimizer的方式创建模型和优化器，再将全局变量传入train_func便不会有这个问题。请问应当如何解决这个问题？

[xxr]

看起来感觉并不是 self.model 引起的问题，能否放上来一份你的最小复现代码呢？