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