-
Notifications
You must be signed in to change notification settings - Fork 0
Open
Description
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
compilemethod 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_generatorcreate a training log for callbacks just like Keras - Modify
fit_generatorto 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
callnotforwardto useforwardfor some forward pass overhead - Create non-generator equivalents (i.e. create Keras's
fit,evaluate, andpredictmethods)
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels