I’m trying joint model training,so my model’s input data is various.
example :
uid_data: [12,3,23,4,54,23,12,43,32...]
feature_data:[1,0,2.3,23,3,0.4...]
text_data:[[1,2,5,3,23,43..],[4,34,54,34,5,...],...]
and many...
for normal way, i need to write these variables one by one to forward function ,like:
class MyModel():
def __init__():
pass
def forward(self,uid_data,uid_mask,feature_data,feature_mask,text_data,text_mask):
pass
#
uid_data=torch.LongTensor(uid)
feature_data=torch.LongTensor(feature)
text_data=torch.LongTensor(text)
...
I want to know if there is an easy way to achieve these like:
class MyModel():
def __init__():
pass
def forward(self,x):
pass
#
X_data=torch.LongTensor(train_X) # element of train_X like: [[12,3,23,4,54,23,12,43,32...] , [1,0,2.3,23,3,0.4...],[[1,2,5,3,23,43..],[4,34,54,34,5,...],...],...]
thanks a lot !