This repository was archived by the owner on Jul 7, 2023. It is now read-only.
This repository was archived by the owner on Jul 7, 2023. It is now read-only.
*help* t2t-decoder uses up both CPU and GPU memory. #836
Description
Description
When I use t2t-decoder to decode from dataset, this command takes up both CPU and GPU memory. I think that t2t-decoder did not use GPU.
When I don't set worker_gpu_memory_fraction, CUDA_ERROE_OUT_OF_MEMORY occurs.
TensorFlow and tensor2tensor versions
tensor flow-gpu==1.5.0 & tensor2tensor==1.6.3
I tried to find out the reason from the source code but failed.Can anyone give me some help? Thanks.