我正在尝试为二进制图像分类问题训练VGG19模型.我的数据集不适合内存,所以我使用批次和.fit_generator函数model.
但是,即使尝试批量训练,我也会收到以下错误:
W tensorflow/core/common_runtime/bfc_allocator.cc:275]尝试分配392.00MiB时内存不足.查看内存状态的日志.
W tensorflow/core/framework/op_kernel.cc:975]资源耗尽:OOM在分配形状的张量时
这是启动训练脚本时关于我的GPU的控制台输出:
Using TensorFlow backend.
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcurand.so locally
Found 20000 images belonging to 2 classes.
Found 5000 images belonging to 2 classes.
I tensorflow/stream_executor/cuda/cuda_gpu_executor.cc:937] successful NUMA node read from SysFS had negative value (-1), …