Find centralized, trusted content and collaborate around the technologies you use most.
Teams
Q&A for work
Connect and share knowledge within a single location that is structured and easy to search.
作为这个线程,我们可以通过将函数 (f:trainer -> bool) 设置为 Trainer 的 stop_triger 来停止迭代循环。但是这样一来,我认为我们不能使用其他扩展,例如使用 stop_trigger=((args.epoch, '10')) 的 LogReport。
所以,我的问题是如何实现作为扩展的提前停止以及如何发送信号以停止训练器从扩展中的迭代。
谢谢。
我在gist上实现了示例代码,并在原始线程上更新了答案。
我注意到stop_trigger最初使用元组表示法(args.epoch, '10'),而不是我们需要更改以传递可调用对象(EarlyStoppingTrigger在上面的示例中)。
stop_trigger
(args.epoch, '10')
EarlyStoppingTrigger