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.

tutorial_9_callback.rst 1.2 kB

12345678910111213141516171819202122232425
  1. ==============================================================================
  2. Callback 教程
  3. ==============================================================================
  4. 在训练时,我们常常要使用trick来提高模型的性能(如调节学习率),或者要打印训练中的信息。
  5. 这里我们提供Callback类,在Trainer中插入代码,完成一些自定义的操作。
  6. 我们使用和 :doc:`/user/quickstart` 中一样的任务来进行详细的介绍。
  7. 给出一段评价性文字,预测其情感倾向是积极(label=1)、消极(label=0)还是中性(label=2),使用 :class:`~fastNLP.Trainer` 和 :class:`~fastNLP.Tester` 来进行快速训练和测试。
  8. 关于数据处理,Loss和Optimizer的选择可以看其他教程,这里仅在训练时加入学习率衰减。
  9. ---------------------
  10. Callback的构建
  11. ---------------------
  12. 创建Callback
  13. 我们可以继承fastNLP :class:`~fastNLP.Callback` 类来定义自己的Callback。
  14. 这里我们先实现一个让学习率线性衰减的Callback。
  15. .. code-block:: python
  16. import fastNLP
  17. class MyCallback(fastNLP.Callback):
  18. def