PyTorch equivalent for Keras sequential model

user_14648732 0
Viewed 522

How to get the perfect copy of this Keras sequential network in PyTorch?

model = tf.keras.models.Sequential([
    tf.keras.layers.Flatten(input_shape=(28, 28)),
    tf.keras.layers.Dense(128, activation='relu'),
    tf.keras.layers.Dense(10)
])
pytorch keras tensorflow machine-learning python
1 Answers

This is a snippet that works for this case:

    model_torch = nn.Sequential(
         nn.Flatten(), 
         nn.Linear(28*28, 128), 
         nn.ReLU(), 
         nn.Linear(128, 10), 
    )
user_24529140
Related