DL学习
文章平均质量分 64
DL相关理论基础学习。
Wilber529
练习时长四年半...
展开
-
TensorRT使用——①安装
介绍TensorRT如何安装原创 2022-09-21 11:06:59 · 1900 阅读 · 1 评论 -
Pytorch多GPU训练程序改造
1 模块导入&FLAG设置import torch.backends.cudnn as cudnnfrom torch.utils.data.distributed import DistributedSamplerimport argparseparser = argparse.ArgumentParser(description='Train segmentation network')parser.add_argument("--local_rank", type=int, d原创 2020-09-11 21:58:13 · 1844 阅读 · 0 评论 -
CenterNet姿势估计decode部分代码解读
代码链接:https://github.com/xingyizhou/CenterNet/blob/1085662179604dd4c2667e3159db5445a5f4ac76/src/lib/models/decode.py#L497代码位置:src/lib/models/decode.py代码注释def multi_pose_decode(heat, wh, kps, r...原创 2020-03-26 18:04:46 · 2787 阅读 · 3 评论 -
Batch Norm理解
BN怎么计算(训练过程)设有BN层,即一个BN层可学习的参数有和。对一个mini-batch,。第一步,计算当前batch的均值和方差(对同一层的所有节点):; 第二步,进行规范化:; 第三步,进行平移和缩放:.BN的推理过程(如何加速BN?)采用在训练过程中收集的全局统计量来代替均值和方差:,.然后将上面两部分带入到中,得到:早期解释《Batch No...原创 2019-10-15 18:44:50 · 390 阅读 · 0 评论