Skip to content

Commit 8f4c90a

Browse files
Revert "[API Compatiblity] Add Optimizer.load_state_dict and Optimizer.zero_g…" (PaddlePaddle#76550)
This reverts commit 820d38a.
1 parent 6996029 commit 8f4c90a

File tree

2 files changed

+0
-47
lines changed

2 files changed

+0
-47
lines changed

python/paddle/optimizer/optimizer.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -512,8 +512,6 @@ def set_state_dict(self, state_dict: dict[str, Tensor]) -> None:
512512
)
513513
var.set_value(state_dict[var_tmp.name])
514514

515-
load_state_dict = set_state_dict
516-
517515
def get_opti_var_name_list(self) -> list[str]:
518516
return self._opti_name_list
519517

@@ -1893,10 +1891,6 @@ def clear_grad(self, set_to_zero: bool = True) -> None:
18931891
for p in param_list:
18941892
p.clear_gradient(set_to_zero)
18951893

1896-
@framework.non_static_only
1897-
def zero_grad(self, set_to_none: bool = True) -> None:
1898-
self.clear_grad(set_to_zero=not set_to_none)
1899-
19001894
@imperative_base.no_grad()
19011895
def minimize(
19021896
self,

test/legacy_test/test_optimizer.py

Lines changed: 0 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -28,17 +28,6 @@
2828
from paddle.io import Dataset
2929

3030

31-
class Model(paddle.nn.Module):
32-
def __init__(self):
33-
super().__init__()
34-
self.conv1 = paddle.nn.Conv2d(1, 64, kernel_size=3, stride=1, padding=1)
35-
36-
def forward(self, x):
37-
x = self.conv1(x)
38-
x = paddle.sum(x)
39-
return x
40-
41-
4231
class TestOptimizerDtype(unittest.TestCase):
4332
'''
4433
The dtype of optimizer should be inferred by parameters, and the learning rate
@@ -175,36 +164,6 @@ def test_with_state_dict(self):
175164
np.testing.assert_array_equal(out_use_state_dict, out_no_state_dict)
176165

177166

178-
class TestNewOptimizerAPI(unittest.TestCase):
179-
def test_zero_grad(self):
180-
paddle.disable_static()
181-
model = Model()
182-
data_input = paddle.randn(64, 1, 28, 28)
183-
data_output = model(data_input)
184-
data_output.backward()
185-
sgd = paddle.optimizer.SGD(
186-
parameters=model.parameters(), weight_decay=0.0
187-
)
188-
sgd.zero_grad()
189-
190-
result = model.conv1.weight.grad
191-
self.assertIsNone(result)
192-
193-
def test_load_state_dict(self):
194-
paddle.disable_static()
195-
theta = paddle.tensor([1.0, 1.0], requires_grad=True)
196-
optim = paddle.optimizer.Optimizer(
197-
parameters=[theta], **{"learning_rate": 1.0}
198-
)
199-
result_before = optim.state_dict()
200-
optim.load_state_dict(result_before)
201-
result_after = optim.state_dict()
202-
self.assertEqual(
203-
set(result_before.keys()),
204-
set(result_after.keys()),
205-
)
206-
207-
208167
if __name__ == '__main__':
209168
paddle.enable_static()
210169
unittest.main()

0 commit comments

Comments
 (0)