Yes, I‘m using 3 target classes (number of occupants 0, 1 or 2) and later under the training I’m using CrossEntropyLoss.
I’m aware of the limitations that you have mentioned but I want to test if it’s a suitable idea for an IoT device.