Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Mimicking Keras Training API #5

Open
8 tasks
abhmul opened this issue Apr 6, 2018 · 2 comments
Open
8 tasks

Mimicking Keras Training API #5

abhmul opened this issue Apr 6, 2018 · 2 comments
Assignees

Comments

@abhmul
Copy link
Owner

abhmul commented Apr 6, 2018

In order to be able to seamlessly swap pyjet models with keras models, pyjet needs to change to use the same API for training and interact the same way with training related keras objects (e.g. callbacks). In order to do so, the following needs to happen:

  • Write a compile method for SLModel to pass in loss functions, optimizers, and metrics into
  • Support multi-input, multi-loss, and multi-output in the same fashion as keras
  • Make fit_generator create a training log for callbacks just like Keras
  • Modify fit_generator to return a history object
  • Replace save_state and load_state with save_weights and load_weights respectively
  • Create save and load_model just like with keras
  • Change impelentable method for the forward pass of the network to call not forward to use forward for some forward pass overhead
  • Create non-generator equivalents (i.e. create Keras's fit, evaluate, and predict methods)
@abhmul abhmul self-assigned this Apr 6, 2018
@abhmul
Copy link
Owner Author

abhmul commented May 24, 2018

#6 Could help with doing this properly

@abhmul
Copy link
Owner Author

abhmul commented May 24, 2018

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

No branches or pull requests

1 participant