From 0904407daabbc00f0b0b277eea4b5bb273936df6 Mon Sep 17 00:00:00 2001 From: Trangle Date: Wed, 15 Dec 2021 13:35:31 +0800 Subject: [PATCH] Update run_classification_criteo_multi_gpu.py change multi_gpu support for newer version --- examples/run_classification_criteo_multi_gpu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/run_classification_criteo_multi_gpu.py b/examples/run_classification_criteo_multi_gpu.py index cff0a617..f25f3409 100644 --- a/examples/run_classification_criteo_multi_gpu.py +++ b/examples/run_classification_criteo_multi_gpu.py @@ -2,7 +2,7 @@ from sklearn.metrics import log_loss, roc_auc_score from sklearn.model_selection import train_test_split from sklearn.preprocessing import LabelEncoder, MinMaxScaler -from tensorflow.python.keras.utils import multi_gpu_model +from tensorflow.python.keras.utils.multi_gpu_utils import multi_gpu_model from deepctr.feature_column import SparseFeat, DenseFeat,get_feature_names from deepctr.models import DeepFM