pytorch+python
错误：
纠正：
self.trace, _ = torch.jit.get_trace_graph(self.model, args=(self.x,))
为：
self.trace, _ = torch.jit._get_trace_graph(self.model, args=(self.x,))
问题解决
相关文章：
相似解决方案
热门标签