訓練模型一直跑不出結果
2019/07/27 09:49 下午
機器學習共學討論版
cathyy
觀看數:67
回答數:4
收藏數:0
ml100-2-d83
ml100-2
我用下載的解答檔案去跑 但是訓練結果跑好幾次 一整天都沒有跑出來 不曉得是程式碼有錯還是真的要跑這麼久?
我的顯卡是 mx130
程式碼是解答複製上來的 這段
results = {}
for i, (use_preact, bz) in enumerate(itertools.product(PRE_ACTIVATE, BATCH_SIZE)):
print("Numbers of exp: %i, use_preact: %s, batch_size: %i" % (i, use_preact, bz))
model = build_mlp(input_shape=x_train.shape[1:], pre_activate=use_preact)
model.summary()
optimizer = keras.optimizers.Adam(lr=LEARNING_RATE)
model.compile(loss="categorical_crossentropy", metrics=["accuracy"], optimizer=optimizer)
model.fit(x_train, y_train,
epochs=EPOCHS,
batch_size=bz,
validation_data=(x_test, y_test),
verbose=0,
shuffle=True)
# Collect results
exp_name_tag = ("exp-%s" % (i))
results[exp_name_tag] = {'train-loss': model.history.history["loss"],
'valid-loss': model.history.history["val_loss"],
'train-acc': model.history.history["acc"],
'valid-acc': model.history.history["val_acc"]}