- 博客(6)
- 资源 (4)
- 收藏
- 关注
原创 Anaconda Prompt中cd命令怎么跨盘符?
Anaconda Prompt中cd命令怎么夸盘符?方法一先输入盘符,再使用cd命令方法二使用如下命令即可跨盘符:cd /d e:\xxx\xxx参考:https://segmentfault.com/a/1190000023369012
2021-03-19 17:56:22 856
原创 torchsummary的使用
安装torchsummarypip install torchsummary例子netE = Encoder(NUM_CHANNELS=3)netG = Generator(NUM_CHANNELS=3, z_dim=1024, z_size=1)netD = Discriminator(NUM_CHANNELS=3, z_dim=1024)netD.to(device)netE.to(device)netG.to(device)summary(netE, input_size=(3,
2020-12-02 09:45:24 902
原创 torch.nn.BCELoss参数详解
定义数学公式为Loss = -w * [p * log(q) + (1-p) * log(1-q)],其中p、q分别为理论标签、实际预测值,w为权重。这里的log对应数学上的ln。PyTorch对应函数为:torch.nn.BCELoss(weight=None, size_average=None, reduce=None, reduction=‘mean’)计算目标值和预测值之间的二进制交叉熵损失函数。有四个可选参数:weight、size_average、reduce、reduction(
2020-11-06 09:35:33 5962
原创 20-11-02 pytorch tensor.no_grad()
引用:https://blog.csdn.net/weixin_43178406/article/details/89517008
2020-11-02 09:14:50 185
原创 20-11-02 Pytorch记录model.eval()
20-11-02 Pytorch记录model.eval()训练完train_datasets之后,model要来测试样本了。在model(test_datasets)之前,需要加上model.eval()因为:orch.onnx.export()导出onnx模型后,利用onnxruntime加载onnx模型后,其输出结果与原始.pth模型的输出结果之间存在很大的差距;通过拆分网络结构,定位到nn.BatchNorm2d()层导致;Batch Normalization和DropoutBatch
2020-11-02 09:05:56 765 3
ImageNet是个非常庞大的数据集,该资源为ImageNet的子集,包含2200多张大象的自然图像
2021-03-19
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人