Inheritance diagram for fastreid.engine.hooks.PeriodicWriter:
Public Member Functions | |
| __init__ (self, writers, period=20) | |
| after_step (self) | |
| after_train (self) | |
 Public Member Functions inherited from fastreid.engine.train_loop.HookBase | |
| before_train (self) | |
| before_step (self) | |
Protected Attributes | |
| _writers | |
| _period | |
Detailed Description
Write events to EventStorage periodically. It is executed every ``period`` iterations and after the last iteration.
Constructor & Destructor Documentation
◆ __init__()
| fastreid.engine.hooks.PeriodicWriter.__init__ | ( | self, | |
| writers, | |||
| period = 20 ) |
Args:
writers (list[EventWriter]): a list of EventWriter objects
period (int):
Definition at line 152 of file hooks.py.
152 def __init__(self, writers, period=20):
153 """
154 Args:
155 writers (list[EventWriter]): a list of EventWriter objects
156 period (int):
157 """
158 self._writers = writers
159 for w in writers:
160 assert isinstance(w, EventWriter), w
161 self._period = period
162
Member Function Documentation
◆ after_step()
| fastreid.engine.hooks.PeriodicWriter.after_step | ( | self | ) |
Called after each iteration.
Reimplemented from fastreid.engine.train_loop.HookBase.
Definition at line 163 of file hooks.py.
163 def after_step(self):
164 if (self.trainer.iter + 1) % self._period == 0 or (
165 self.trainer.iter == self.trainer.max_iter - 1
166 ):
167 for writer in self._writers:
168 writer.write()
169
◆ after_train()
| fastreid.engine.hooks.PeriodicWriter.after_train | ( | self | ) |
Called after the last iteration.
Reimplemented from fastreid.engine.train_loop.HookBase.
Definition at line 170 of file hooks.py.
170 def after_train(self):
171 for writer in self._writers:
172 writer.close()
173
174
Member Data Documentation
◆ _period
◆ _writers
The documentation for this class was generated from the following file:
- smreid/fastreid/engine/hooks.py
