江户香葱饼的博客

问题描述：在工程上同时使用了Tensorflow框架模型与pytorch框架模型。全部模型大小加起来1个G左右，显存11G，但是运行时报错：CUDA out of memoryRuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 10.91 GiB total capacity; 856.79 MiB already allocated; 17.38 MiB free; 71.21 MiB cached)

1、类别不平衡问题的定义：类别不平衡（class-imbalance）就是指分类任务中，不同类别的训练数据数目差别很大的情况。2、类别不平衡问题的影响：假设训练样例中正例数目m+，负例数目m-，那么观测几率为 m+/m- 。设一个样本预测为正例的概率为p，则预测为负例的概率为1-p，则分类器的预测几率为p/1-p。通常情况下，我们假设训练集是真实样本总体的无偏采样，那么当预测几...