nbox.models
This is experimental and not ready for production use, can be removed without notice
Classes
classModelTrainer
[source]
function__init__
[source]
functioninit
[source]functiontrain_model
[source]Trains the model
functioninfer
[source]Runs the inference for any inputs that the user passes
functionsave_checkpoint
[source]Returns the path to the checkpoint
functionload_checkpoint
[source]Loads the checkpoint and assigns the relevant attributes like model
functionlog
[source]Logs the metrics to NBX
functioncheckpoint
[source]functionget_checkpoint
[source]functiondeploy
[source]functiontrain
[source]Run the training part of the pipeline
functionserve
[source]Run the inference part of the pipeline
function_update_run_config
[source]Updates the run config with the key and value