@@ -43,35 +43,38 @@ def save_persistables(model_dict, dirname='save_dir', optimizers=None):
43
43
optimizers(fluid.Optimizer|list(fluid.Optimizer)|None): The optimizers to be saved
44
44
45
45
Returns:
46
+ None
46
47
47
48
Examples:
49
+
48
50
.. code-block:: python
49
- ptb_model = PtbModel(
51
+
52
+ ptb_model = PtbModel(
50
53
hidden_size=hidden_size,
51
54
vocab_size=vocab_size,
52
55
num_layers=num_layers,
53
56
num_steps=num_steps,
54
57
init_scale=init_scale)
55
- sgd = fluid.optimizer.SGD(learning_rate=0.01)
56
- x_data = np.arange(12).reshape(4, 3).astype('int64')
57
- y_data = np.arange(1, 13).reshape(4, 3).astype('int64')
58
- x_data = x_data.reshape((-1, num_steps, 1))
59
- y_data = y_data.reshape((-1, 1))
60
- init_hidden_data = np.zeros(
58
+ sgd = fluid.optimizer.SGD(learning_rate=0.01)
59
+ x_data = np.arange(12).reshape(4, 3).astype('int64')
60
+ y_data = np.arange(1, 13).reshape(4, 3).astype('int64')
61
+ x_data = x_data.reshape((-1, num_steps, 1))
62
+ y_data = y_data.reshape((-1, 1))
63
+ init_hidden_data = np.zeros(
61
64
(num_layers, batch_size, hidden_size), dtype='float32')
62
- init_cell_data = np.zeros(
65
+ init_cell_data = np.zeros(
63
66
(num_layers, batch_size, hidden_size), dtype='float32')
64
- x = to_variable(x_data)
65
- y = to_variable(y_data)
66
- init_hidden = to_variable(init_hidden_data)
67
- init_cell = to_variable(init_cell_data)
68
- dy_loss, last_hidden, last_cell = ptb_model(x, y, init_hidden,
67
+ x = to_variable(x_data)
68
+ y = to_variable(y_data)
69
+ init_hidden = to_variable(init_hidden_data)
70
+ init_cell = to_variable(init_cell_data)
71
+ dy_loss, last_hidden, last_cell = ptb_model(x, y, init_hidden,
69
72
init_cell)
70
- dy_loss.backward()
71
- sgd.minimize(dy_loss)
72
- ptb_model.clear_gradient()
73
- param_path = "./my_paddle_model"
74
- fluid.dygraph.save_persistables(ptb_model.state_dict(), dirname=param_path, sgd)
73
+ dy_loss.backward()
74
+ sgd.minimize(dy_loss)
75
+ ptb_model.clear_gradient()
76
+ param_path = "./my_paddle_model"
77
+ fluid.dygraph.save_persistables(ptb_model.state_dict(), dirname=param_path, sgd)
75
78
"""
76
79
if isinstance (model_dict , collections .OrderedDict ):
77
80
_save_var_to_file (model_dict , optimizers , dirname , None )
@@ -95,13 +98,15 @@ def load_persistables(dirname='save_dir'):
95
98
optimizer dict: The optimizer
96
99
97
100
Examples:
98
- .. code-block:: python
99
- my_layer = layer(fluid.Layer)
100
- param_path = "./my_paddle_model"
101
- sgd = SGDOptimizer(learning_rate=1e-3)
102
- param_dict, optimizer_dict = fluid.dygraph.load_persistables(my_layer.parameters(), param_path)
103
- param_1 = param_dict['PtbModel_0.w_1']
104
- sgd.load(optimizer_dict)
101
+
102
+ .. code-block:: python
103
+
104
+ my_layer = layer(fluid.Layer)
105
+ param_path = "./my_paddle_model"
106
+ sgd = SGDOptimizer(learning_rate=1e-3)
107
+ param_dict, optimizer_dict = fluid.dygraph.load_persistables(my_layer.parameters(), param_path)
108
+ param_1 = param_dict['PtbModel_0.w_1']
109
+ sgd.load(optimizer_dict)
105
110
106
111
"""
107
112
return _load_var_from_file (dirname )
0 commit comments