TESTING PURPOSE FOR SEGMENTATIOn Need help for GPU Optiimizing Currently able to perform 1 batch only via Colabs