Skip to content

Latest commit

 

History

History
25 lines (14 loc) · 1019 Bytes

README.md

File metadata and controls

25 lines (14 loc) · 1019 Bytes
Raw

PyTorch-Models

  • Implementing PyTorch models as seen in the below figures

Sequential 1: Model with no hidden layer

  • Train for 10 epochs (output is softmax)

    alt text

Sequential 2: Model with single hidden layer

  • Use sigmoid in hidden layer, train for 20 epochs

    alt text

    Sequential 3: Deep model

  • Use ReLU in hidden layer, train for 20 epochs

    alt text

    Functional: Complex deep model

  • Use sigmoid for layers within the blue box and use ReLU in the remaining hidden layers, train for 25 epochs

    alt text