When I run this model using python:
def build_network(): models = [] model = Sequential() model.add(Dense(64, input_dim=122)) model.add(Dense(32)) model.add(Activation('relu')) model.add(Dropout(.15)) model.add(Dense(32)) model.add(Activation('relu')) model.add(Dropout(.15)) model.add(Dense(32)) model.add(Activation('relu')) model.add(Dropout(.15)) model.add(Dense(5)) model.add(Activation('softmax')) model.compile(loss='categorical_crossentropy', optimizer='adam', metrics=['accuracy']) return model NN = build_network() early_stopping = EarlyStopping(monitor='val_loss', min_delta=0, patience=3, verbose=0, mode='auto') NN.fit(x=train, y=target, epochs=100, validation_split=0.1, batch_size=128, callbacks=[early_stopping]) |
I receive this error:
InternalError: Blas GEMM launch failed : a.shape=(128, 122), b.shape=(122, 64), m=128, n=64, k=122 [[{{node dense_6/MatMul}}]] |