I tried the following:
def hybrid_forward(self, F, in1, in2, in3, in4, in5, in6, in7, in8):
In the main code:
net = MyNetwork()
args = (a1, a2, a3, a4, a5, a6, a7, a8)
outY = net(*args)
But, that gives me the following error:
AssertionError: HybridBlock input must be (nested) list of Symbol or NDArray, but got 32 of type <class 'int'>
Am I doing it wrong?
P.S. Interestingly, I noticed that the above code does work if the net.hybridize() line is commented out, but, gives the error mentioned above if the line is not commented out.