mousecpn/MachineLearning_HW

老师请问pytorch中有类似的callback函数吗?

Opened this issue · 2 comments

tensorfow 中有回调机制,可以使得loss值达到一定值后,调用函数停止训练,防止迭代次数过大导致过拟合的现象。请问pytorch中有类似的回调函数吗?

tensorfow 中有回调机制,可以使得loss值达到一定值后,调用函数停止训练,防止迭代次数过大导致过拟合的现象。请问pytorch中有类似的回调函数吗?

你说的是early stopping吧,官方接口没有,大概是要自己写。
多使用百度https://www.baidu.com/s?ie=UTF-8&wd=pytorch%20early%20stopping

收到,谢谢!打扰老师了