code: def activate_train_mode_for_all_layers(model: Callable) -> Callable:
      model.trainable = False  # type: ignore
      model = partial(
          model, training=True
      )  # Note this affects batchnorm etc also
      return model
  
label_desc: Use image node_id 1 for calling a global function with example usage: activate_train_mode_for_all_layers(model) and returns: model