You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Add EarlyStopping and ModelCheckpoint callbacks, example
Bug Fixes and Other Changes
Use float64 in metric to prevent nan/inf loss when calculating logloss
API changes
DCN add a new parameter: cross_parameterization: string, "vector" or "matrix", way to parameterize the cross network.if set to "matrix" then it will be DCN-M
Add EarlyStopping and ModelCheckpoint in deepctr_torch.callbacks
Fix RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [128]], which is output 0 of SelectBackward, is at version 2; expected version 1 instead. Hint: enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True) in : #88 #98 #90 #102