Pytorch
jmucvm
这个作者很懒,什么都没留下…
展开
-
论文阅读 | Ghostnet网络
简介paper: https://arxiv.org/abs/1911.11907github:https://github.com/huawei-noah/ghostnetGhostnet(CVPR2020)是华为提出的一种轻量级网络,结构类似mobilenetV3,具有速度快、精度较高特点。论文中主要贡献是:提出Ghost Module。Ghost Module如图二所示,...原创 2020-03-09 22:02:53 · 694 阅读 · 0 评论 -
pytorch | tensor维度中使用 None
None可以在所处维度中多一维,具体实现。import torchtensor = torch.randn(3, 4)print('tensor size:', tensor.size())tensor_1 = tensor[:, None]print('tensor_1 size:', tensor_1.size())tensor_2 = tensor[:, :, None]p...原创 2019-12-30 21:03:46 · 16878 阅读 · 2 评论 -
Pytorch | dataloader 多线程下numpy每个线程随机种子都一样解决方案。
问题描述pytorch的Dataloader用于加载数据。在num_works >1时, 每个线程中numpy.random产生的随机数一样,也就是随机种子相同。random 和 torch两个模块的随机数不会出现这种情况。DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, batch_...原创 2019-12-04 21:21:00 · 2120 阅读 · 2 评论