Join TensorFlow at Google I/O, May 11-12 Register now


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.