BooTurbo
-
Linux/Ubuntu系统下使用VS Code配置C/C++开发环境
在Ubuntu下，使用VS Code来编辑代码或进行开发非常方便，下面记录一下如何配置gcc/g++编译器和GDB调试工具。 准备工作： 1. 安装VS Code，过程略。 2. 为VS Code安装C/C++ Extension Pack 扩展组件，其他插件会附带安装 3. Ubuntu系统自带g++和gdb，查看一下 配置环境: VS Code …
-
PyTorch保存模型断点以及加载断点继续训练
在训练神经网络时，用到的数据量可能很大，训练周期较长，如果半途中断了训练，下次从头训练就会很费时间，这时我们就想断点续训。 一、神经网络模型的保存，基本两种方式：1. 保存完整模型model， torch.save(model, save_path) 2. 只保存模型的参数， torch.save(model.state_dict(), sav…
-
F.binary_cross_entropy_with_logits函数与F.binary_cross_entropy函数的关系（二分类问题）
二元交叉熵（Binary cross entropy）是二分类中常用的损失函数，它可以衡量两个概率分布的距离，二元交叉熵越小，分布越相似，其公式如下： 我们用jupyter notebook举例解释一下， 假设 1个图像样本由神经网络处理后的输出是 size 10×4 的tensor，随机生成一个tensor， 使用Sigmoid对该tenso…
