ProgressBar¶
-
class
ProgressBar(total: int, num_epochs: int)¶ Bases:
objectSimple progress bar, used for visual representation durign training.
Example:
pb = ProgressBar(5000, 20) pb.print_bar(500, 2, 0.976)
Output:
Epoch: 1/20 |===========...................| 25.5% loss: 0.976
-
__init__(total: int, num_epochs: int) → None¶ Initalizes the object.
- Parameters
total (int) – total number of iteration per epoch
num_epochs (int) – total number of epochs
Methods
__init__(total, num_epochs)Initalizes the object.
print_bar(iteration, epoch, loss)Used to print each iteration of progress bar
-
print_bar(iteration: int, epoch: int, loss: float) → None¶ Used to print each iteration of progress bar
- Parameters
iteration (int) – current iteration
epoch (int) – current epoch
loss (float) – current loss
- Returns
Nothing
-
