如何在 PyTorch 中獲得這個 Keras 順序網路的完美副本?
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)
])
uj5u.com熱心網友回復:
這是適用于這種情況的片段:
model_torch = nn.Sequential(
nn.Flatten(),
nn.Linear(28*28, 128),
nn.ReLU(),
nn.Linear(128, 10),
)
轉載請註明出處,本文鏈接:https://www.uj5u.com/houduan/406898.html
標籤:
