ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more


Loads data and retrains the model.

Used in the notebooks

Used in the tutorials

train_data A instance of audio_dataloader.DataLoader class.
model_spec Specification for the model.
validation_data Validation DataLoader. If None, skips validation process.
batch_size Number of samples per training step. If use_hub_library is False, it represents the base learning rate when train batch size is 256 and it's linear to the batch size.
epochs Number of epochs for training.
model_dir The location of the model checkpoint files.
do_train Whether to run training.
train_whole_model Boolean. By default, only the classification head is trained. When True, the base model is also trained.

An instance based on AudioClassifier.