tensorflow/models · error · ValueError

Unsupported mode, only support `train`

Error message

Unsupported mode, only support `train`

What it means

Error "Unsupported mode, only support `train`" thrown in tensorflow/models.

Source

Thrown at official/projects/edgetpu/nlp/run_mobilebert_edgetpu_train.py:110

        max_to_keep=5,
        step_counter=runner.current_step,
        checkpoint_interval=20000,
        init_fn=None)

  controller = orbit.Controller(
      trainer=runner,
      evaluator=runner,
      global_step=runner.current_step,
      strategy=distribution_strategy,
      steps_per_loop=experiment_params.orbit_config.steps_per_loop,
      summary_dir=os.path.join(model_dir, 'train'),
      eval_summary_dir=os.path.join(model_dir, 'eval'),
      checkpoint_manager=checkpoint_manager)

  if FLAGS.mode == 'train':
    controller.train(steps=experiment_params.orbit_config.total_steps)
  else:
    raise ValueError('Unsupported mode, only support `train`')

if __name__ == '__main__':
  tfm_flags.define_flags()
  app.run(main)

View on GitHub (pinned to e006f5f0d5)

When it happens

Trigger: Thrown at official/projects/edgetpu/nlp/run_mobilebert_edgetpu_train.py:110 when the library encounters an invalid state.

Common situations: See trigger scenarios.

AI-assisted analysis of tensorflow/models@e006f5f0d5 (2026-08-24). Data as JSON: /api/errors/bfe37970a311687d. Report an issue: GitHub.