深度学习
lzh_zyy
这个作者很懒,什么都没留下…
展开
-
k8s的docker解决修改shm的大小问题记录
项目场景: 基于docker跑单机多卡的分布式训练 问题描述 在训练的时候报shm的大小不够的问题,想通过添加shm-size参数改变大小,发现不生效 解决方案: spec: volumes: - name: dshm emptyDir: medium: Memory volumeMounts: - name: dshm mountPath: /dev/shm ...原创 2022-03-12 14:38:59 · 1750 阅读 · 1 评论 -
pytorch中BatchNorm2d的方法测试
pytorch中BatchNorm2d的方法测试代码测试 代码测试 batchnormal是对一个batch_size中的每个channel进行归一化 from torch import nn import torch m = nn.BatchNorm2d(3) input = torch.randn(4,3,2,2) output = m(input) input tensor([[[[ 0.4306, -0.7709], [ 0.6851, 0.1494]],转载 2020-05-09 14:47:28 · 364 阅读 · 0 评论