界面已本地化;技能介绍与说明文档按所选语言由 AI 自动翻译,代码、命令、路径和专业术语保留原文,以原作者文档为准。
探索技能
数据与研究 · SKILL.md

training

training

【数据与研究】Train LaneNet on prepared data, manage checkpoints, and debug training runs.

独立 GitHub 快照 · 自动分类 · 以原作者文档为准

技能说明

GitHub

name: training description: "Train LaneNet on prepared data, manage checkpoints, and debug training runs." metadata: disco-role: operating disable-model-invocation: true license: Apache 2.0

LaneNet Training

Use this sub-skill when the task is to train or resume LaneNet, choose the config-driven backbone, inspect TensorBoard/checkpoints, or debug why a run produced no snapshots or unstable loss.

Route here for

  • Training on prepared TFRecords from TuSimple or custom lane data.
  • Selecting the front end through MODEL.FRONT_END (bisenetv2 default, vgg supported).
  • Single-GPU vs multi-GPU behavior through TRAIN.MULTI_GPU.ENABLE.
  • Restore-from-snapshot runs, warm-up/poly learning rate behavior, mIoU summaries, and loss inspection.
  • Understanding where snapshots, TensorBoard logs, and training logs are written.

Route elsewhere

Required context before acting

  1. Confirm the prepared data root contains LaneNet TFRecords, especially tusimple_train.tfrecords and, for multi-GPU validation, tusimple_val.tfrecords.
  2. Confirm the runtime is CUDA-capable with TensorFlow 1.15 compatibility. That is the validated training path in this repo.
  3. Confirm whether the user wants from-scratch training or restore-from-snapshot behavior.
  4. Confirm the backbone and device mode. The current source uses config-driven backbone selection rather than a command-line --net flag.
  5. Use the bundled wrapper in scripts/train_lanenet_tusimple.py; it resolves REPO_ROOT_PATH placeholders, checks the training data size, and refuses to start the no-iteration tiny-data trap.

Key outputs

  • Checkpoints under TRAIN.MODEL_SAVE_DIR/<front_end>_lanenet/.
  • TensorBoard summaries under TRAIN.TBOARD_SAVE_DIR/<front_end>_lanenet/.
  • Training log file under LOG.SAVE_DIR/.
  • A copied model_train_config.json beside the TensorBoard event files.

References and wrapper

我的收藏