我有一个简单的线性回归模型,其中自变量是年份(1970年至今)。当我将输入数据集中在零(即从x中减去平均值)时,我的模型运行得很好,得到了最佳拟合线。但是如果我不把数据集中起来,这个模型就会有无限的损失: tf.keras.layers.Input(shape=(1,)we no longer get nan\inf
我有cuda8.0.61,tensorflow_gpu版本和keras。我在224*224图像数据上训练一个20层的keras模型当我在终端上运行nvidia -smi时,我发现内存正在耗尽,并且计算util的百分比更小。当我尝试拟合模型时,机器变得非常慢。我知道,为了使用gpu并在设备之间切换,我应该使用以下代码: with K.tf.device('