Shipping a Neural Network on iOS with CoreML, PyTorch, and React Native
Really great end-to-end write up.Just a few things: in general case it's better not to use MSE after sigmoid due to slow convergence.And "logits" variable is not logits actually, it's probabilities. Logits is what you have before applying sigmoid activation.