(self, epoch_current)
| 135 | return self._time_used / self._iter |
| 136 | |
| 137 | def eta(self, epoch_current): |
| 138 | epoch_current += 1 # since counter starts from 0 |
| 139 | time_per_epoch = self._time_total / epoch_current |
| 140 | return time_per_epoch * (self._epoch_total - epoch_current) |
| 141 | |
| 142 | def update_stats(self, true, pred, loss, lr, time_used, params, **kwargs): |
| 143 | assert true.shape[0] == pred.shape[0] |