سلام دوستان، یه شبکه کوچیک (دولایه کانولوشن و دوتا پولینگ و یک کلاسبند) دارم. وقتی که شبکه برای آموزش روی gpu برده میشه تقریبا تمام رم gpu-2G رو میگیره و در این صورت نمیتونم بچ سایز رو زیاد بگیرم.
نکته1: آپشن های allow_soft_placement=True و allow_growth =True و هم تست کردم، اینا باعث میشن که تنسورفلو مینیمم رم لازم رو اشغال کنه ولی باز وقتی بچ رو بزرگ میگیرم خطای حافظه می ده.
نکته 2: دیتا ست رو هم با تعریف یک صف از فایل tfrecord میخونم.
پیشاپیش ممنون از راه حلتون.