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
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
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:
compile
method for SLModel to pass in loss functions, optimizers, and metrics intofit_generator
create a training log for callbacks just like Kerasfit_generator
to return a history objectcall
notforward
to useforward
for some forward pass overheadfit
,evaluate
, andpredict
methods)The text was updated successfully, but these errors were encountered: