Hi,thank you for your amazing job
I try to use your work to visualize my own model
When I run :
my model
import hiddenlayer as hl
hl.build_graph(model, torch.zeros([1, 3, 224, 224]))
but I got:
AttributeError: module 'torch.jit' has no attribute '_get_trace_graph'
Could you tell me how can I slove it?
Thank you
Hi,thank you for your amazing job I try to use your work to visualize my own model When I run :
my model
import hiddenlayer as hl hl.build_graph(model, torch.zeros([1, 3, 224, 224])) but I got: AttributeError: module 'torch.jit' has no attribute '_get_trace_graph' Could you tell me how can I slove it? Thank you