2023年3月28日
Understanding transform.Normalize( )
摘要： 如果自己的数据集和 imageNet 数据集相差较大，需要计算自己数据集的 mean 和 std。 Normalize 让模型可以更快的收敛。 参考：https://discuss.pytorch.org/t/understanding-transform-normalize/21730/17 ht 阅读全文
posted @ 2023-03-28 18:53 Zenith_Hugh 阅读(96) 评论(0) 推荐(0)
深度学习学习率下降策略
摘要： 参考： 【1】梯度下降学习率的设定策略 【2】CosineAnnealingLR-余弦退火调整学习率 【3】图解 9 种Torch中常用的学习率调整策略 【4】炼丹总结 阅读全文
posted @ 2023-03-28 18:35 Zenith_Hugh 阅读(6) 评论(0) 推荐(0)
计算pytorch 模型的推理时间和 模型大小
摘要： 脚本 import torch import time import sys sys.path.insert(0, sys.path[0]+"/../") from cls_models.model import ClsModel if __name__ == '__main__': model_n 阅读全文
posted @ 2023-03-28 17:52 Zenith_Hugh 阅读(26) 评论(0) 推荐(0)