老师请问pytorch中有类似的callback函数吗?
Opened this issue · 2 comments
hakunacc commented
tensorfow 中有回调机制,可以使得loss值达到一定值后,调用函数停止训练,防止迭代次数过大导致过拟合的现象。请问pytorch中有类似的回调函数吗?
mousecpn commented
tensorfow 中有回调机制,可以使得loss值达到一定值后,调用函数停止训练,防止迭代次数过大导致过拟合的现象。请问pytorch中有类似的回调函数吗?
你说的是early stopping吧,官方接口没有,大概是要自己写。
多使用百度https://www.baidu.com/s?ie=UTF-8&wd=pytorch%20early%20stopping
hakunacc commented
收到,谢谢!打扰老师了