pytorch
biubiubiu888
这个作者很懒,什么都没留下…
展开
-
nvidia driver install
1. 禁止 nouveaunouveau 是系统自带的一个显示驱动程序,需要先将其禁用.`cat /lib/modprobe.d/dist-blacklist.conf注释掉#blacklist nvidiafb在# blacklist nvidiafb下面加上blacklist nouveauoptions nouveau modeset=0reboot重启2. 重建镜像m...原创 2019-12-06 16:14:03 · 795 阅读 · 0 评论 -
nn.utils.rnn.pack_padded_sequence 与 nn.utils.rnn.pad_packed_sequence
nn.utils.rnn.pack_padded_sequence ,nn.utils.rnn.pad_packed_sequence以及参考https://www.cnblogs.com/lindaxin/p/8052043.htmlimport torchimport torch.nn as nnfrom torch.autograd import Variablef...原创 2019-03-13 21:29:56 · 2279 阅读 · 0 评论 -
pytorch BN函数说明
pytorch中的BN继承自:class _BatchNorm(Module):这个类的代码:class _BatchNorm(Module): def __init__(self, num_features, eps=1e-5, momentum=0.1, affine=True, track_running_stats=True):...原创 2019-05-13 13:00:40 · 12471 阅读 · 1 评论