keras知识点(持续更新)

我使用的keras是基于Tensorflow后端的框架(建议大家使用这个)

1、keras 调用gpu方法

如果linux服务器中keras 没有默认gpu操作的话，那么在代码前面加入这三行命令即可选择调用的gpu：

os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"（其中0.1是选择所调用的gpu）

gpu_options = tf.GPUOptions(allow_growth=True)

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

2、keras打印模型结构

1、model.summary()：打印出模型概况，它实际调用的是keras.utils.print_summary

参考：https://blog.csdn.net/jacke121/article/details/78991786