[問題] Keras 每次結果不同

[f496328mm (123)]

我目前使用 keras 的 vgg16 進行建模，但是每次結果都不一樣，

即使我設定 seed，shuffle 也設 false

code 如下，想問問看有沒有什麼解決方法，

我上網 google ，有可能是因為 GPU 運算的關係? 那要如何解?

謝謝

PS: 環境

ubuntu 16.04
python3
gtx-1070





model_vgg16_conv = VGG16(weights=None, include_top=False)
input = Input(shape=(rows,cols,3),name = 'image_input')

output_vgg16_conv = model_vgg16_conv(input)

x = Flatten(name='flatten')(output_vgg16_conv)
x = Dense(1024, activation='relu', name='fc1')(x)
x = Dense(1024, activation='relu', name='fc2')(x)
x = Dense(2, activation='softmax', name='predictions')(x)

model = Model(input=input, output=x)
sgd = SGD(lr=1e-4, momentum=0.2, nesterov=True)
model.compile(sgd, loss='binary_crossentropy')


random.seed(100)
set_random_seed(100)
np.random.seed(1337)
model.fit(sub_train_x,
          sub_train_y,
          batch_size=64,
          epochs=2,
          verbose=1,
          #validation_split=0.2,
          shuffle=False)

--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 114.36.50.200
※ 文章網址: https://www.ptt.cc/bbs/Python/M.1504674952.A.F26.html
推 hl4: 可以參考一下https://github.com/fchollet/keras/issues/2743
謝謝

我也有找到這篇，不過還是不太行，發文者最後也沒有解決

其實網路上很多討論，大家都有類似的問題，然後解決方法都不同@@

目前在一個一個測試中，不過情況不樂觀，

希望有相關經驗的高手，分享經驗 > <
※ 編輯: f496328mm (114.36.50.200), 09/06/2017 19:01:29
→ hl4: 他有找到可行的做法阿，只是要重開 ipynb 而已