I have already build my network and now I would like to pass input to network. My forward() method looks:
def forward(self, input, index=None):
if index is not None:
#Do something
else:
#Do something else
And the way I pass input to network is:
output = net(input, 1)
And this leads to problem as index is still ‘None’ even I pass value ‘1’ to network.
After searching online, it seems that forward() method can only accept input in Variable type.
Therefore it seems alright after I modified the codes as following:
output = net(input, Variables(torch.IntTensor([1])))
However, is there any simpler way of doing this? Thank you for your time.