Pytorch 记录 每一次迭代的 数据加载、模型计算 等时长

Pytorch 记录 每一次迭代的 数据加载、模型计算 等时长

使用 time 库如何记录pytorch处理每一批次时间消耗

可结合 tqdm 工具使用 效果更佳

import time

class Process:
    """
     Pytorch 记录 每一次迭代的 数据加载、模型计算 等时长
    """
    def __init__(self) -> None:
        pass

    def record_time