mindquantum.utils.TwoLoopsProgress

mindquantum.utils.TwoLoopsProgress#

class mindquantum.utils.TwoLoopsProgress(n_outer_loop: int, n_inner_loop: int, outer_loop_name: str = 'Epoch', inner_loop_name: str = 'Batch')#

A progress bar of task with two level for loops.

Parameters:

  • n_outer_loop (int) – Total loop size of outer loop.

  • n_inner_loop (int) – Total loop size of inner loop.

  • outer_loop_name (str) – The name that will be shown as the title of outer loop progress bar. Default: "Epoch".

  • inner_loop_name (str) – The name that will be shown as the title of inner loop progress bar. Default: "Batch".

Examples

>>> import time
>>> from mindquantum.utils import TwoLoopsProgress
>>> with TwoLoopsProgress(3, 100) as progress:
>>>     for ep in range(3):
>>>         for batch in range(100):
>>>             progress.update_inner_loop(batch)
>>>             time.sleep(0.01)
>>>         progress.update_outer_loop(ep)
update_inner_loop(loop_idx: int)#

Update inner loop progress bar.

Parameters:

loop_idx (int) – The index of inner loop.

update_outer_loop(loop_idx: int)#

Update outer loop progress bar.

Parameters:

loop_idx (int) – The index of outer loop.

Contents