主要用于对比和记录模型复现过程中的各个步骤精度对齐情况
类 ReprodLogger 用于记录和报错复现过程中的中间变量
主要方法为
- add(key, val)：添加key-val pair
- remove(key)：移除key
- clear()：清空字典
- save(path)：保存字典
类 ReprodDiffHelper 用于对中间变量进行检查，主要为计算diff
主要方法为
- load_info(path): 加载字典文件
- compare_info(info1:dict, info2:dict): 对比diff
- report(diff_threshold=1e-6,path=None): 可视化diff，保存到文件或者到屏幕