

Train

train

def train(base_model_name, n_classes, samples, image_dir, batch_size, epochs_train_dense, epochs_train_all, learning_rate_dense, learning_rate_all, dropout_rate, job_dir, img_format, existing_weights, multiprocessing_data_load, num_workers_data_load, decay_dense, decay_all, **kwargs)