苏剑林(Jianlin Su)

Results 390 comments of 苏剑林(Jianlin Su)

算了,给你个大致例子吧 ```python for layer in model.layers: if xxxx: layer.trainable = False model = Model(model.inputs, model.outputs) ``` 具体原理请搜索“keras freeze layers”

这个很显然就是没用gpu训练。而为什么没用gpu训练,那是因为没有搭好tf的gpu环境。至于怎么搭好tf的gpu环境,那就请自行搜索了。

可以用`tf.test.is_gpu_available()`测试是否搭好了tf的gpu环境。

> > 可以用`tf.test.is_gpu_available()`测试是否搭好了tf的gpu环境。 > > 感谢您的快速恢复。 > 能否告知您那边的TensorFlow-gpu、cuda和cudnn的版本号? > 我对应的配置一下,谢谢 我tf用1.14或1.15,对应的cuda/cudnn版本号直接网上就可以找到了,cuda是10.x吧

模型的搭建部分代码可以发一下吗?目前看不出什么问题。最好简化一下,提供一个可复现的代码,谢谢反馈。

试试在模型的最后输出接个Lambda层: ```python output = Lambda(lambda x: x)(output) ```

用回python2试试。 你这信息量这么少,叫别人怎么回答....