You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

training.md 928 B

3 years ago
123456789101112131415161718
  1. # Training
  2. From the previous tutorials, you may now have a custom model and data loader.
  3. You are free to create your own optimizer, and write the training logic: it's
  4. usually easy with PyTorch, and allow researchers to see the entire training
  5. logic more clearly.
  6. One such example is provided in [tools/plain_train_net.py](https://github.com/facebookresearch/detectron2/blob/master/tools/plain_train_net.py).
  7. We also provide a standarized "trainer" abstraction with a
  8. [minimal hook system](../modules/engine.html#detectron2.engine.HookBase)
  9. that helps simplify the standard types of training.
  10. You can use
  11. [SimpleTrainer().train()](../modules/engine.html#detectron2.engine.SimpleTrainer)
  12. which does single-cost single-optimizer single-data-source training.
  13. Or use [DefaultTrainer().train()](../modules/engine.html#detectron2.engine.defaults.DefaultTrainer)
  14. which includes more standard behavior that one might want to opt in.

No Description