因此,我试图用多个数据集来获取CNN,当我添加足够的数据时(例如当我添加多个集合作为一个或当我尝试添加具有超过一百万个样本的数据集时),它会抛出ResourceExhaustedError . 至于指令here,我尝试添加from keras.backend.tensorflow_backend import set_session import tensorflow as tf config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.3 set_session(tf.Session(config=config))我的代码,但这不...
from keras.preprocessing.image import ImageDataGenerator from keras.models import Sequential from keras.layers import Conv2D, MaxPooling2D from keras.layers import Activation, Dropout, Flatten, Dense from keras import backend as K# dimensions of our images. img_width, img_height = 150, 150train_data_dir = 'flowers/train' validation_data_dir = 'flowers/validation' nb_train_samples = 2500 nb_validat...