簡體   English   中英

深度學習無法運行model.fit函數

[英]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']))

你有一個錯字錯誤,請嘗試保換metricmetrics

另外,您還應將損失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.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM