search

DeepRec-AI / HybridBackend

A high-performance framework for training wide-and-deep recommender systems on heterogeneous cluster
Apache License 2.0
156 stars 30 forks source link

init_from_checkpoint throw Exception when using hb.keras.Model #120

Closed karterotte closed 1 year ago

karterotte commented 1 year ago

Current behavior

python code like:

tf.feature_column.embedding_column(
      tf.feature_column.categorical_column_with_identity(
          key=offset_name,
          num_buckets=dict_size,
          default_value=0),
      combiner='sqrtn',
      dimension=dimension,
      initializer=glorot_uniform(),
      ckpt_to_load_from=ckpt_to_load_from,
      tensor_name_in_ckpt=tensor_name_in_ckpt

It throw Exception 'tuple' object has no attribute 'is_compatible_with' when ckpt_to_load_from was setted. It's OK when ckpt_to_load_from and tensor_name_in_ckpt are None

Willing to contribute

Yes

francktcheng commented 1 year ago

This issue has been resolved in (#123 )