سلام
من از کراس با بکند تنسور استفاده میکنم .تصویر ورودیم دوهزاز عکس 200*200 ،من وقتی از cpu استفاده میکنم سرعت کمی داره ولی خطای حافظه نمیده الان از gpu 1050 ti روی لپ تاپ استفاده میکنم که 4 گیگ . خود سیستم 24 گیگ رم داره و بعد از دو بار تکرار خطای OOM میده
ایا راه حلی هست که این اتفاق نیوفته
در ضمن batch_size هم 8 گذاشتم ولی با این حال باز این خطارو میده