Skip to content

Commit a785a83

Browse files
committed
update the example with the latest API
1 parent 3eef539 commit a785a83

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

python/paddle/fluid/tests/book/word2vec/no_test_word2vec_new_api.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ def train(use_cuda, is_sparse, save_path):
9494
place = fluid.CUDAPlace(0) if use_cuda else fluid.CPUPlace()
9595

9696
def event_handler(event):
97-
if isinstance(event, fluid.EndPass):
97+
if isinstance(event, fluid.Event.END_EPOCH):
9898
avg_cost = trainer.test(reader=paddle.dataset.imikolov.test(
9999
word_dict, N))
100100

@@ -106,10 +106,9 @@ def event_handler(event):
106106

107107
trainer = fluid.Trainer(
108108
partial(inference_network, is_sparse),
109-
optimizer=fluid.optimizer.SGD(learning_rate=0.001),
110-
place=place,
111-
event_handler=event_handler)
112-
trainer.train(train_reader, 100)
109+
fluid.optimizer.SGD(learning_rate=0.001),
110+
place=place)
111+
trainer.train(train_reader, 100, event_handler)
113112

114113

115114
def infer(use_cuda, save_path):

0 commit comments

Comments
 (0)