سلام وقتتون بخیر.

اگر تسکهایی که انجام میدین با eager execution میتونه کار کنه، میتونین این گزینه رو فعال کنین. کافیه توی اون خطی از کد که دستور مقابل داخلش نوشته شده، کامنتش رو بردارین
tf.compat.v1.enable_eager_execution()

میتونین یه فانکشن بنویسین که داخلش دیتا رو لود میکنین و یه سری پیش پردازش انجام میدین. این فانکشن رو باید با @tf.function دکوریت کنین (کافیه که عبارت @tf.function رو بالای اون تابع موقع تعریف کردنش قرار بدین)

نکته شاید مهمی هم که هست اینه که تنسورفلو ۱ واقعا قدیمی شده و به نظرم بهتره که از API نسخه ۲ استفاده کنین. هم راحت تره و هم Lazy Loading رو به خوبی پشتیبانی میکنه.