You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
from tensorflow.python import pywrap_tensorflow
reader = pywrap_tensorflow.NewCheckpointReader(latest_checkpoint)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print(reader.get_tensor(key))
I want export the value of embedding variables, and I test it in nvtf successfully. But in deeprec, the value is [], an empty list.
The text was updated successfully, but these errors were encountered:
I want export the value of embedding variables, and I test it in nvtf successfully. But in deeprec, the value is [], an empty list.
The text was updated successfully, but these errors were encountered: