Detectron2 开始训练 | 八

作者|facebookresearch 编译|Flin 来源|Github

训练

从前面的教程中,你现在可能已经有了一个自定义模型和数据加载器。

你可以自由创建自己的优化器,并编写训练逻辑：使用PyTorch通常很容易,并且使研究人员可以看到整个训练逻辑更清晰并具有完全控制权。 tools/plain_train_net.py中提供了一个这样的示例。 (tools/plain_train_net.py:https://github.com/facebookresearch/detectron2/blob/master/tools/plain_train_net.py)

我们还提供了标准化的"trainer"抽象,最小hook系统(https://detectron2.readthedocs.io/modules/engine.html#detectron2.engine.HookBase) ,这有助于简化标准的训练类型。

你可以使用 SimpleTrainer().train() ,它为单次成本单优化器单数据源训练提供了最小的抽象。内置train_net.py脚本使用 DefaultTrainer().train(),它包含一个人们可能希望选择的更多标准默认行为。这也意味着它不太可能支持你在研究过程中可能想要的一些非标准行为。

• SimpleTrainer().train():