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
Currently newsrec models only support epoch training and the smoke-tests with epoch = 1 take minutes for each notebook.
With a simple tweak to base_model.fit(), we can introduce steps parameters that constraint the number of batches to train.
Then, we can set steps = 1 to the smoke test which will expediate the test time.
Expected behavior with the suggested feature
Use steps = 1 to all the smoke tests of the newsrec models.
The text was updated successfully, but these errors were encountered:
Description
Currently
newsrec
models only supportepoch
training and the smoke-tests withepoch = 1
take minutes for each notebook.With a simple tweak to
base_model.fit()
, we can introducesteps
parameters that constraint the number of batches to train.Then, we can set
steps = 1
to the smoke test which will expediate the test time.Expected behavior with the suggested feature
Use
steps = 1
to all the smoke tests of the newsrec models.The text was updated successfully, but these errors were encountered: