shenweichen / DeepCTR

Easy-to-use,Modular and Extendible package of deep-learning based CTR models .
https://deepctr-doc.readthedocs.io/en/latest/index.html
Apache License 2.0
7.54k stars 2.21k forks source link

I'm using this model with cpu, so I'm getting an error. #538

Open byunsunyoung opened 10 months ago

byunsunyoung commented 10 months ago

I'm using the DCNMix model. I trained and saved the model using gpu. I'm using this model with cpu, so I'm getting an error. Why is there an error? Please solve the problem.

code: df_test_seg_dup_pred = loaded_model.predict(df_test_seg_dup_model_input, 32) error: InvalidArgumentError Traceback (most recent call last) Cell In[42], line 2 1 with tf.device("CPU:0"): ----> 2 df_test_seg_dup_pred = loaded_model.predict(df_test_seg_dup_model_input, 32)

File /opt/conda/lib/python3.8/site-packages/keras/utils/traceback_utils.py:67, in filter_traceback..error_handler(*args, **kwargs) 65 except Exception as e: # pylint: disable=broad-except 66 filtered_tb = _process_traceback_frames(e.traceback) ---> 67 raise e.with_traceback(filtered_tb) from None 68 finally: 69 del filtered_tb

File /opt/conda/lib/python3.8/site-packages/tensorflow/python/eager/execute.py:54, in quick_execute(op_name, num_outputs, inputs, attrs, ctx, name) 52 try: 53 ctx.ensure_initialized() ---> 54 tensors = pywrap_tfe.TFE_Py_Execute(ctx._handle, device_name, op_name, 55 inputs, attrs, num_outputs) 56 except core._NotOkStatusException as e: 57 if name is not None:

InvalidArgumentError: Graph execution error:

Detected at node 'model/linear0sparse_emb_cat_t1a030300010/embedding_lookup' defined at (most recent call last): File "/opt/conda/lib/python3.8/runpy.py", line 194, in _run_module_as_main