HugeCTR icon indicating copy to clipboard operation
HugeCTR copied to clipboard

[Question] tensorflow 1.15 sok example

Open MichoChan opened this issue 2 years ago • 2 comments

run_sok_horovod.py use GradientTape, but tf1.15 has no GradientTape func? can provide 1.15 example with sok?

MichoChan avatar Sep 07 '23 14:09 MichoChan

Hi @MichoChan , you can use https://github.com/NVIDIA-Merlin/HugeCTR/blob/main/sparse_operation_kit/sparse_operation_kit/experiment/test/function_test/tf1/lookup/lookup_sparse_distributed_test.py to have a try

kanghui0204 avatar Sep 08 '23 01:09 kanghui0204

Hi @MichoChan , you can use https://github.com/NVIDIA-Merlin/HugeCTR/blob/main/sparse_operation_kit/sparse_operation_kit/experiment/test/function_test/tf1/lookup/lookup_sparse_distributed_test.py to have a try

ok, i take a try to test

MichoChan avatar Sep 08 '23 07:09 MichoChan