event.py 410 字节
Newer Older
Y
Yu Yang 已提交
1 2
"""
All training events.
3 4 5 6 7 8 9 10 11 12 13

There are:

* BeginTraining
* EndTraining
* BeginIteration
* EndIteration
* BeginPass
* EndPass

TODO(yuyang18): Complete it!
Y
Yu Yang 已提交
14
"""
Y
Yu Yang 已提交
15 16 17 18 19 20 21 22 23 24 25 26
__all__ = ['EndIteration']


class EndIteration(object):
    """
    Event On One Batch Training Complete.
    """

    def __init__(self, pass_id, batch_id, cost):
        self.pass_id = pass_id
        self.batch_id = batch_id
        self.cost = cost