Bug: Tensorflow not responding when run code

[Nanatsume]

What happened?

i'm try move to run code with gpu on wsl2 ( ubuntu ) , and i installed and check everything is okay.
but when i run code on VS code with remote WSL , output is not happen and keep me to terminal instead.
when i run on cpu is okay and the same version tensorflow. It very bad for me because i use cpu to train is very slow TT

Relevant code

https://github.com/Nanatsume/DogClassification/blob/e1b50a3737fccbad8c3c4c4f0842e118e86e30d9/train.py

Relevant log output

-

tensorflow_hub Version

other (please specify)

TensorFlow Version

other (please specify)

Other libraries

tensorflow 2.13.0

Python Version

3.x

OS

Linux