【发布时间】:2017-01-19 12:56:52
【问题描述】:
我尝试使用 vgg-19 模型实现样式转换。我使用 tensorflow 作为后端通过现有的 vgg-19 keras 模型加载了权重。首先,我在我的笔记本上尝试了它,虽然在 tf 0.11.0 上运行很慢,但它运行良好。然后我用 gpus 切换到我的 AWS 实例,它的版本为 0.12.1,并尝试了相同的脚本:它在 jupyter notebook 中没有产生任何错误,而是冻结内核并在我尝试计算输出时强制它重新启动一个特定的层,除了第一层的输入。我安装了 0.11.0 的 gpu 版本——同样的问题,虽然两个版本都通过了矩阵乘法测试。
我还尝试在 0.12.1 上运行某人的脚本,使用为 0.11.0 编写的 vgg-16,会话由于层尺寸错误而崩溃,尽管它在 cpu 上的 0.11.0 上运行良好。 tf 0.11.0 的 GPU 版本 - 相同的冻结和重启。
除了 python 控制台中的日志之外,还有 tensorflow 写入的日志吗?我还没有找到关于他们的任何提及
【问题讨论】:
标签: tensorflow