[英]Deep learning cannot run model.fit function
我正在使用Cifar-10數據集,並且正在嘗試通過使用keras庫進行轉移學習。 我的代碼在這里-https://github.com/YanaNeykova/Cifar-10在運行時
model.fit(X_train, y_train, batch_size=32, epochs=10,
verbose=1, callbacks=[checkpointer],validation_split=0.2, shuffle=True)
我收到錯誤消息(在文件中可見),因此無法繼續進行操作。 我還嘗試了另外從keras導入模型函數,但是我再次得到了相同的結果-無法識別函數模型。 有人可以建議我如何進行嗎? 提前謝謝了!
錯誤
---------------------------------------------------------------------------
ValueError Traceback (most recent call last)
<ipython-input-11-977cb2a1e5d6> in <module>()
1 model.fit(X_train, y_train, batch_size=32, epochs=10,
----> 2 verbose=1, callbacks=[checkpointer],validation_split=0.2, shuffle=True)
/usr/local/lib/python3.6/dist-packages/keras/engine/training.py in fit(self, x, y, batch_size, epochs, verbose, callbacks, validation_split, validation_data, shuffle, class_weight, sample_weight, initial_epoch, steps_per_epoch, validation_steps, **kwargs)
1008 else:
1009 ins = x + y + sample_weights
-> 1010 self._make_train_function()
1011 f = self.train_function
1012
/usr/local/lib/python3.6/dist-packages/keras/engine/training.py in _make_train_function(self)
517 updates=updates,
518 name='train_function',
--> 519 **self._function_kwargs)
520
521 def _make_test_function(self):
/usr/local/lib/python3.6/dist-packages/keras/backend/tensorflow_backend.py in function(inputs, outputs, updates, **kwargs)
2742 msg = 'Invalid argument "%s" passed to K.function with TensorFlow backend' % key
2743 raise ValueError(msg)
-> 2744 return Function(inputs, outputs, updates=updates, **kwargs)
2745
2746
/usr/local/lib/python3.6/dist-packages/keras/backend/tensorflow_backend.py in __init__(self, inputs, outputs, updates, name, **session_kwargs)
2573 raise ValueError('Some keys in session_kwargs are not '
2574 'supported at this '
-> 2575 'time: %s', session_kwargs.keys())
2576 self._callable_fn = None
2577 self._feed_arrays = None
ValueError: ('Some keys in session_kwargs are not supported at this time: %s', dict_keys(['metric']))
你有一個錯字錯誤,請嘗試保換metric
與metrics
另外,您還應將損失binary_crossentropy
糾正為caegorical_crossentropy
model.compile(loss='caegorical_crossentropy', optimizer='adam',
metrics=['accuracy'])
您正在使用以下行似乎不支持的度量關鍵字參數:
model.compile(loss='binary_crossentropy', optimizer='adam',
metric=['accuracy'])
嘗試將其刪除,看看是否可行。 您的模型可能不支持它。
另外,我注意到您可能在損失函數loss='caegorical_crossentropy'
的名稱中有錯別字...但是我想這是另一個問題。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.