/xor-layers-pytorch

Using PyTorch nn Layers to learn XOR

Primary LanguageJupyter Notebook

XOR in PyTorch

This is a minimal implementation of XOR in PyTorch and Jupyter

Neural Network Model

Input

Dimension: 2

Hidden Layer

Dimension: 2

Activation Function: ReLU

Output

Dimension: 1

Activation Function: Sigmoid

Loss

Cross Entropy

Optimization

SGD with momentum