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