code: def train_model(model):
      model.compile(
          optimizer=tf.keras.optimizers.Adam(1e-2),
          loss=tf.keras.losses.SparseCategoricalCrossentropy(
              from_logits=True
          ),
          metrics=[
              tf.keras.metrics.SparseCategoricalAccuracy(
                  name="accuracy"
              )
          ],
      )
      (x_train, y_train), _ = get_dataset()
      model.fit(x_train, y_train, batch_size=128, epochs=3)
  
imports: import os
  import shutil
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_model_optimization.python.core.common.keras.compression.algorithms import epr
label_desc: Use image node_id 3 for calling a global function with example usage: train_model(model) without return types