logo
Loading...

訓練模型一直跑不出結果 - Cupoy

我用下載的解答檔案去跑 但是訓練結果跑好幾次 一整天都沒有跑出來 不曉得是程式碼有錯還是真的要跑這麼...

ml100-2-d83,ml100-2

訓練模型一直跑不出結果

2019/07/27 01:49 PM
機器學習新手論壇
cathyy
觀看數:0
回答數: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"]}