update

Scheaven

2021-09-18 291deeb1fcf45dbf39a24aa72a213ff3fd6b3405

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
B  
©r_(    ã@s*dd
lZ
ddlmZmZ
Gdd„dƒZd
S)éN)ÚListÚTuplec@sfeZ
dZd
Zdedœ
dd„Zdeedœdd    „Zd
d„Zedœ
d d„Z    edœ
dd„Z
dd„Zdd„ZdS)Ú HistoryBufferz†
    Track a series of scalar values and provide access to smoothed values over a
    window or the global average of the series.
    é@B)
Ú
max_lengthcCs|
|_g|_
d
|_d
|_dS)zÔ
        Args:
            max_length: maximal number of values that can be stored in the
                buffer. When the capacity of the buffer is exhausted, old
                values will be removed.
        r
N)Ú_max_lengthÚ_dataÚ_countÚ_global_avg)Úselfr©rú./utils/history_buffer.pyÚ__init__s


zHistoryBuffer.__init__N)ÚvalueÚ    iterationcCsf|d
kr|j}t
|jƒ
|jkr*|j d¡

|j |
|f¡

|jd7_|j|
|j|j7_d
S)zÁ
        Add a new scalar value produced at certain iteration. If the length
        of the buffer exceeds self._max_length, the oldest element will be
        removed from the buffer.
        Nr
é
)r    ÚlenrrÚpopÚappendr
)rrrrrr Úupdates




zHistoryBuffer.updatec

Cs|jd
dS)zE
        Return the latest scalar value added to the buffer.
        éÿÿÿÿr
)
r)
rrrr Úlatest)szHistoryBuffer.latest)
Úwindow_sizecCs t 
d
d„|j|
d…Dƒ
¡
S)zU
        Return the median of the latest `window_size` values in the buffer.
        c
Ssg|]}
|
d‘qS)
r
r)Ú.0Ú
xrrr ú
<listcomp>3sz(HistoryBuffer.median.<locals>.<listcomp>N)ÚnpÚmedianr)rrrrr r/szHistoryBuffer.mediancCs t 
d
d„|j|
d…Dƒ
¡
S)zS
        Return the mean of the latest `window_size` values in the buffer.
        c
Ssg|]}
|
d‘qS)
r
r)rrrrr r9sz%HistoryBuffer.avg.<locals>.<listcomp>N)rÚmeanr)rrrrr Úavg5szHistoryBuffer.avgc


Cs|jS)
z™
        Return the mean of all the elements in the buffer. Note that this
        includes those getting removed due to limited buffer storage.
        )
r
)
rrrr Ú
global_avg;szHistoryBuffer.global_avgc


Cs|jS)
z`
        Returns:
            list[(number, iteration)]: content of the current buffer.
        )
r)
rrrr ÚvaluesBszHistoryBuffer.values)
r)
N) Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚintrÚfloatrrrrr r!rrrr rsr)ÚnumpyrÚtypingrrrrrrr Ú<module>s