search

NVIDIA-Merlin / HugeCTR

HugeCTR is a high efficiency GPU framework designed for Click-Through-Rate (CTR) estimating training
Apache License 2.0
950 stars 200 forks source link

[Question] tensorflow 1.15 sok example #418

Open MichoChan opened 1 year ago

MichoChan commented 1 year ago

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

kanghui0204 commented 1 year ago

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

MichoChan commented 1 year ago

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