Open WeeHyongTok opened 3 years ago
preds = model(sent_id, mask) # compute the loss between actual and predicted values loss = cross_entropy(preds, labels)
When train is invoked , I see the following error - AttributeError: 'tuple' object has no attribute 'dim'
get model predictions for the current batch
When train is invoked , I see the following error - AttributeError: 'tuple' object has no attribute 'dim'