摘要: 在用 PyTorch 训练不同模型时,需要编写很多重复的逻辑。一个工程的代码无可避免地变得越来越长,难以管理。PyTorch Lightning 作为一个对 PyTorch 二次封装的框架,能让训练逻辑的编写像堆积木一样秩序井然。 虽然叫做 lightning,这个库的学习成本并不低。好在一但熟悉, 阅读全文
posted @ 2024-01-14 20:05 倒地 阅读(1030) 评论(0) 推荐(1)
摘要: F1 score 与 平均精度 mAP 可以衡量分类模型的性能。 首先先看精确率与召回率的概念。 精确率与召回率 对于某方概率极低的逻辑回归(例如某种罕见病症的确诊),单纯的准度(是否能正确判断病人是否得病)并不足够。毕竟,若算法只会给出 negative 的判断也能拥有高准度。 真阳性(TP),假 阅读全文
posted @ 2024-01-14 13:57 倒地 阅读(344) 评论(0) 推荐(0)