From e956b236d58f928704c2b5d9a70ca6c466ac6a8c Mon Sep 17 00:00:00 2001 From: LielinJiang Date: Wed, 22 Apr 2020 06:50:40 +0000 Subject: [PATCH] update readme --- examples/style-transfer/README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/style-transfer/README.md b/examples/style-transfer/README.md index d91610a..84b57bd 100644 --- a/examples/style-transfer/README.md +++ b/examples/style-transfer/README.md @@ -176,8 +176,8 @@ model.prepare(optimizer, style_loss) ```python # 使用内容图像和风格图像获取内容特征和风格特征 -content_fetures = model.test(content) -style_features = model.test(style) +content_fetures = model.test_batch(content) +style_features = model.test_batch(style) ``` @@ -191,7 +191,7 @@ feats = style_features + [content_fetures[-2]] # 训练5000个step，每500个step画一下生成的图像查看效果 steps = 5000 for i in range(steps): - outs = model.train(target, feats) + outs = model.train_batch(target, feats) if i % 500 == 0: print('iters:', i, 'loss:', outs[0]) -- GitLab