File tree Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Expand file tree Collapse file tree 1 file changed +8
-2
lines changed Original file line number Diff line number Diff line change
1
+ import sys
1
2
import paddle .v2 as paddle
2
3
3
4
@@ -104,7 +105,9 @@ def main():
104
105
parameters = paddle .parameters .create (cost )
105
106
106
107
# define optimize method and trainer
107
- optimizer = paddle .optimizer .Adam (learning_rate = 1e-4 )
108
+ optimizer = paddle .optimizer .Adam (
109
+ learning_rate = 5e-5 ,
110
+ regularization = paddle .optimizer .L2Regularization (rate = 1e-3 ))
108
111
trainer = paddle .trainer .SGD (cost = cost ,
109
112
parameters = parameters ,
110
113
update_equation = optimizer )
@@ -125,8 +128,11 @@ def main():
125
128
def event_handler (event ):
126
129
if isinstance (event , paddle .event .EndIteration ):
127
130
if event .batch_id % 10 == 0 :
128
- print "Pass %d, Batch %d, Cost %f, %s" % (
131
+ print "\n Pass %d, Batch %d, Cost %f, %s" % (
129
132
event .pass_id , event .batch_id , event .cost , event .metrics )
133
+ else :
134
+ sys .stdout .write ('.' )
135
+ sys .stdout .flush ()
130
136
131
137
# start to train
132
138
trainer .train (
You can’t perform that action at this time.
0 commit comments