core.utils.trainutils.LossInfo
core.utils.trainutils.LossInfo(
epoch,
len_dataset,
batch_size,
mode='Train',
print_freq=1,
)Attributes
| Name | Description |
|---|---|
| batch_size | |
| epoch | |
| len_dataset | |
| loss | |
| loss_tracker | |
| mode | |
| print_freq | |
| t | |
| times |
Methods
| Name | Description |
|---|---|
| get_avg | |
| print_info | |
| update | |
| update_timer |
get_avg
core.utils.trainutils.LossInfo.get_avg()print_info
core.utils.trainutils.LossInfo.print_info(batch_idx)update
core.utils.trainutils.LossInfo.update(loss, n)update_timer
core.utils.trainutils.LossInfo.update_timer(timer_mode='Dataloader')