@B_Whitewind thanks for the hint! I only want to do binary image classification to start out with :comfy: