search

waleedka / hiddenlayer

Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.
MIT License
1.79k stars 266 forks source link

how can we use this tool with a model that have multiple parts in input? #61

Open chg0901 opened 4 years ago

chg0901 commented 4 years ago

For example, my model takes two inputs like the following coding shows

outputs = net(inputs, tr_angles)

chg0901 commented 4 years ago

For example, my model takes two inputs like the following coding shows

outputs = net(inputs, tr_angles)

I have tried graph = hl.build_graph(net, inputs, tr_angles) and graph = hl.build_graph(net, (inputs, tr_angles)) and graph = hl.build_graph(net, [inputs, tr_angles])

luciaL commented 4 years ago

Hello, have you solved thsi question? I have the same question as you.

checche commented 2 years ago

For models that require two inputs (e.g. outputs = net(input1, input2)), graph = hl.build_graph(net, (input1, input2)) worked fine in v0.3.