Attend the Women in ML Symposium on December 7 Register now


Stay organized with collections Save and categorize content based on your preferences.

Loads data and train the model for object detection.

Used in the notebooks

Used in the tutorials

train_data Training data.
model_spec Specification for the model.
validation_data Validation data. If None, skips validation process.
epochs Number of epochs for training.
batch_size Batch size for training.
train_whole_model Boolean, False by default. If true, train the whole model. Otherwise, only train the layers that are not match model_spec.config.var_freeze_expr.
do_train Whether to run training.

An instance based on ObjectDetector.