提交 e956b236 编写于 作者: L LielinJiang

update readme

上级 c52e6900
...@@ -176,8 +176,8 @@ model.prepare(optimizer, style_loss) ...@@ -176,8 +176,8 @@ model.prepare(optimizer, style_loss)
```python ```python
# 使用内容图像和风格图像获取内容特征和风格特征 # 使用内容图像和风格图像获取内容特征和风格特征
content_fetures = model.test(content) content_fetures = model.test_batch(content)
style_features = model.test(style) style_features = model.test_batch(style)
``` ```
...@@ -191,7 +191,7 @@ feats = style_features + [content_fetures[-2]] ...@@ -191,7 +191,7 @@ feats = style_features + [content_fetures[-2]]
# 训练5000个step,每500个step画一下生成的图像查看效果 # 训练5000个step,每500个step画一下生成的图像查看效果
steps = 5000 steps = 5000
for i in range(steps): for i in range(steps):
outs = model.train(target, feats) outs = model.train_batch(target, feats)
if i % 500 == 0: if i % 500 == 0:
print('iters:', i, 'loss:', outs[0]) print('iters:', i, 'loss:', outs[0])
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册