[ 仅 API 调用方式不一致 ]torch.optim.Optimizer.state_dict

torch.optim.Optimizer.state_dict

torch.optim.Optimizer.state_dict()

paddle.optimizer.Optimizer.state_dict

paddle.optimizer.Optimizer.state_dict()

两者功能一致,但调用方式不一致,具体如下:

转写示例

# PyTorch 写法
optim = torch.optim.Optimizer([theta], defaults={"learning_rate": 1.0})
result = optim.state_dict()

# Paddle 写法
optim = paddle.optimizer.Optimizer(parameters=[theta], **{"learning_rate": 1.0})
result = optim.state_dict()