[fix] Fatal error: When restoring from kv files using insert_or_assign, the recovered data will be the same first data in vector.

MoFHeka · rhdong · commit 29df8b356db7 · 2023-10-28T13:22:58.000+08:00
diff --git a/tensorflow_recommenders_addons/dynamic_embedding/core/kernels/lookup_impl/lookup_table_op_cpu.h b/tensorflow_recommenders_addons/dynamic_embedding/core/kernels/lookup_impl/lookup_table_op_cpu.h
@@ -294,7 +294,7 @@ class TableWrapperDefault final : public TableWrapperBase<K, V> {
     ValueType value_vec;
     value_vec.reserve(value_dim);
     for (int64 j = 0; j < value_dim; j++) {
-      value_vec.push_back(*value);
+      value_vec.push_back(*(value + j));
     }
     return table_->insert_or_assign(*key, value_vec);
   }

Original file line number	Diff line number	Diff line change
`@@ -294,7 +294,7 @@ class TableWrapperDefault final : public TableWrapperBase<K, V> {`
`294`	`294`	`ValueType value_vec;`
`295`	`295`	`value_vec.reserve(value_dim);`
`296`	`296`	`for (int64 j = 0; j < value_dim; j++) {`
`297`		`- value_vec.push_back(*value);`
	`297`	`+ value_vec.push_back(*(value + j));`
`298`	`298`	`}`
`299`	`299`	`return table_->insert_or_assign(*key, value_vec);`
`300`	`300`	`}`