pytorch

pytorch 实现二分类交叉熵逆样本频率权重

通常，由于类别不均衡，需要使用weighted cross entropy lo...

5 年前 0 0 42

pytorch 使用半精度模型部署的操作

背景 pytorch作为深度学习的计算框架正得到越来越多的应用. 我们除了在模型...

5 年前 0 0 60

pytorch 如何使用float64训练

pytorch默认使用单精度float32训练模型，原因在于：使用float...

5 年前 0 0 59

pytorch中Schedule与warmup_steps的用法说明

1. lr_scheduler相关 lr_scheduler = WarmupL...

5 年前 0 0 39

解决pytorch中的kl divergence计算问题

偶然从pytorch讨论论坛中看到的一个问题，KL divergence dif...

5 年前 0 0 45

Pytorch数据读取之Dataset和DataLoader知识总结

目录一、前言二、Dataset 三、DatasetLoader 四、将Dat...

5 年前 0 0 52

pytorch 实现变分自动编码器的操作

本来以为自动编码器是很简单的东西，但是也是看了好多资料仍然不太懂它的原理。先把代...

5 年前 0 0 46

pytorch 实现计算 kl散度 F.kl_div()

先附上官方文档说明：https://pytorch.org/docs/stabl...

5 年前 0 0 52

PyTorch 如何自动计算梯度

在PyTorch中，torch.Tensor类是存储和变换数据的重要工具，相比于...

5 年前 0 0 47

Pytorch 中net.train 和 net.eval的使用说明

在训练模型时会在前面加上： model.train() 在测试模型时在前面使用：...

5 年前 0 0 56

pytorch中的model.eval()和BN层的使用

看代码吧~ class ConvNet(nn.module): def __in...

5 年前 0 0 54

聊聊pytorch测试的时候为何要加上model.eval()

Do need to use model.eval() when I test?...

5 年前 0 0 55