Skip to content

Conversation

akshayvkt
Copy link

  1. Updated the deprecated modules (training_epoch_end and validation_epoch_end) to the new modules (on_train_epoch_end and on_validation_epoch_end).

  2. Removed 'outs argument from both functions because the model fails to train with the outs argument in the function.

The model now successfully trains with these changes. I'm certain that change #1 is needed, I'm not sure about change #2 but updating that let me train the model and not error out.

1. Updated the deprecated modules (training_epoch_end and validation_epoch_end) to on_train_epoch_end and on_validation_epoch_end. 

2. Removed 'outs argument from both functions because the model fails to train with the outs argument in the function.

The model now successfully trains with these changes. I'm certain change rasbt#1 is needed, not sure about change rasbt#2 but updating that let me train the model and not error out.
@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant