What should I do if the input to my model not only consists of Tensors but also python lists?
That is,
model = Net()
add_graph(model, (Tensor1, Tensor2, list))
In this case, I get errors. How can I pass a list as an input to add_graph
?
What should I do if the input to my model not only consists of Tensors but also python lists?
That is,
model = Net()
add_graph(model, (Tensor1, Tensor2, list))
In this case, I get errors. How can I pass a list as an input to add_graph
?