| 160 | |
| 161 | |
| 162 | class LossMeter(object): |
| 163 | def __init__(self): |
| 164 | self.reset() |
| 165 | |
| 166 | def reset(self,): |
| 167 | self.losses = {} |
| 168 | |
| 169 | def update_iter(self, losses): |
| 170 | for key, value in losses.items(): |
| 171 | self.add(key, value) |
| 172 | |
| 173 | def add(self, name, loss): |
| 174 | if name not in self.losses: |
| 175 | self.losses[name] = AverageMeter() |
| 176 | self.losses[name].update(loss) |
| 177 | |
| 178 | def get(self, name): |
| 179 | if name not in self.losses: |
| 180 | return 0 |
| 181 | return self.losses[name] |
nothing calls this directly
no outgoing calls
no test coverage detected