我正在尝试遵循本教程,但我无法训练第一个网络。它崩溃并显示以下消息:
Process finished with exit code -1073740791 (0xC0000409)
并且不会抛出错误,所以我不知道该再谷歌什么
我使用的是 Pycharm 社区版和 python 3.7。
我还用谷歌搜索了不同的教程,但它们都非常相似。我已经完成了构建模型的所有工作,但训练无法开始。本教程使用了 model.fit_generate 函数,但引发了错误,所以我使用了 model.fit 函数。我是否错误地使用了 fit 函数?
这是我的代码:
# baseline model for the dataset
import sys
from matplotlib import pyplot
from keras.utils import to_categorical
from keras.models import Sequential
from keras.layers import Conv2D
from keras.layers import MaxPool2D
from keras.layers import Dense
from keras.layers import Flatten
from keras.optimizers import SGD
from keras.preprocessing.image import ImageDataGenerator
# define cnn model
def define_model():
model = Sequential()
model.add(Conv2D(32, (3, 3), activation='relu', kernel_initializer='he_uniform', padding='same',
input_shape=(200, 200, 3)))
model.add(MaxPool2D((2, 2)))
model.add(Flatten())
model.add(Dense(128, activation='relu', kernel_initializer='he_uniform'))
model.add(Dense(1, activation='sigmoid'))
# compile model
opt = SGD(lr=0.001, momentum=0.9)
model.compile(optimizer='adam', loss='binary_crossentropy', metrics=['accuracy'])
model.summary()
return model
# run the test harness for evaluating a model
def run_test_harness():
# define model
model = define_model()
# create data generator
datagen = ImageDataGenerator(rescale=1.0 / 255.0)
# prepare iterators
train_it = datagen.flow_from_directory('data_dogs_vs_cats/train', class_mode='binary', batch_size=64,
target_size=(200, 200))
test_it = datagen.flow_from_directory('data_dogs_vs_cats/val', class_mode='binary', batch_size=64,
target_size=(200, 200))
# fit model
history = model.fit(train_it, steps_per_epoch=len(test_it), validation_data=test_it, validation_steps=len(test_it),
epochs=20, verbose=0)
# evaluate model
_, acc = model.evaluate_generator(test_it, steps=len(test_it), verbose=0)
print('> %.3f' % (acc * 100.0))
# entry point
run_test_harness()
我遇到了这个问题,原因是 PyCharm 直接从内部运行代码时隐藏了错误。尝试从命令行/终端运行代码,您将看到哪些错误导致了此崩溃。
当 cuda 的 bin 目录中缺少 DLL 或代码中存在其他错误时,可能会发生这种情况。
在我的情况下,我的 cuda bin 文件夹(C:\Program Files\NVIDIA GPU 计算工具包\CUDA)中缺少以下 DLL: