框架版本:
- Tensorflow:2.9.0
- PyTorch:1.11.0
算子 | TensorFlow | PyTorch | MindSpore | Paddle |
---|---|---|---|---|
sgd | tf.keras.optimizers.SGD{:target="_blank"} | torch.optim.SGD | ||
momentum | tf.keras.optimizers.SGD | torch.optim.SGD | ||
adagrad | tf.keras.optimizers.Adagrad | torch.optim.Adagrad | ||
adadelta | tf.keras.optimizers.Adadelta | torch.optim.Adadelta | ||
rmsprop | tf.keras.optimizers.RMSprop | torch.optim.RMSprop | ||
centered_rmsprop | tf.raw_ops.ApplyCenteredRMSProp | / | ||
adam | tf.keras.optimizers.Adam | torch.optim.Adam | ||
adamax | tf.keras.optimizers.Adamax | torch.optim.Adamax |