1 模型断点续训,optimizer.step()报错:RuntimeError Expected all tensors to be on the same device, but found cu…
- Pytorch
- 2023-05-08
1 Tensorboard 1.1 安装和卸载 使用以下命令安装 pip install tensorboard 卸载命令 pip uninstall tensorboard 1.2 Tensorbo…
- Pytorch
- 2023-04-18
转载自https://chenglu.me/blogs/pytorch-model-modification-part1,少量修改,如侵权,请联系我进行删除。 在深度学习网络构建时,我们可能需要对Py…
- Pytorch
- 2023-03-28
1 torch.topk 形式 torch.topk(input, k, dim=None, largest=True, sorted=True, *, out=None) 或者 Tensor.top…
- Pytorch
- 2022-12-15
1 batch size对模型训练的影响 小的batch size引入的数据集的数据量较小,随机性越大,在部分情况下模型难以收敛,影响模型训练效率。 而在合理的范围内,越大的batch size本质上…
- Pytorch
- 2022-12-09
1 torch.reshape 形式 torch.reshape(input, shape) 功能 返回一个与输入张量数据和元素数相同的,但是形状为shape的张量。 参数 input:需要被重新定义…
- Pytorch
- 2022-12-09
1 torch.cat函数 形式 torch.cat(tensors, dim=0, *, out=None) 功能 在指定的维度连接给定序列的张量,所有张量必须具有相同的形状(连接维度除外)或者为空…
- Pytorch
- 2022-12-08
Pytorch中,unsqueeze和squeeze为两个对应的反操作函数,其中,unsqueeze主要用于为输入张量升维,squeeze主要用于给张量降维,两者的具体用法可以参考下文。 1 unsq…
- Pytorch
- 2022-12-08
本文对Pytorch中创建随机数Tensor的一些常用方法进行总结,主要包括 randn randn_like rand rand_like randint randint_like 1 torch.…
- Pytorch
- 2022-12-08
1 torch.nn.Conv2d torch.nn.Conv2d主要对输入Tensor应用2D卷积。 比如输入(N,C_{in},H,W)维度的Tensor,则输出(N,C_{out},H,W)的T…
- Pytorch
- 2022-12-07
1 Pytorch训练网络时出现_pickle.UnpicklingError: pickle data was truncated错误 今天重新在一个大的数据集上重新训练一个模型时,在第二个epoc…
- Pytorch
- 2022-11-07