hiddenlayer
hiddenlayer copied to clipboard
how can we use this tool with a model that have multiple parts in input?
For example, my model takes two inputs like the following coding shows
outputs = net(inputs, tr_angles)
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])
Hello, have you solved thsi question? I have the same question as you.
For models that require two inputs (e.g. outputs = net(input1, input2)),
graph = hl.build_graph(net, (input1, input2)) worked fine in v0.3.