Yasunori Yamamoto
Yasunori Yamamoto
Thanks. When trying the script at my PC, the following error occurred. I have already installed the jax module by `pip install jax` . ``` 2023-12-15 16:51:48.224877: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT...
This happens when there is no GPUs, but I am not sure how to workaround this.
Thanks, it may be reasonable to try at a GPU machine. Just for a reference, I put the full stack trace below. ``` /Users/yayamamo/.pyenv/versions/3.12.2/lib/python3.12/site-packages/torch/cuda/amp/grad_scaler.py:126: UserWarning: torch.cuda.amp.GradScaler is enabled, but CUDA...