Suspect not very fit with GPU, use CPU only with
tf.config.set_visible_devices([], 'GPU')
or
with tf.device('/cpu:0'):
solve the problem. It can prosess as expect.
Topic:
Machine Learning & AI
SubTopic:
General
Tags: