摘要: 1、指定某个显卡 可以在文件开头加入如下代码: import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "1, 2" # 使用第二,三块GPU(从0开始) 禁用GPU 阅读全文
posted @ 2022-08-06 00:31 理想几岁 阅读(2479) 评论(0) 推荐(0) 编辑
摘要: 在神经网络训练中,好的权重 初始化会加速训练过程。 下面说一下kernel_initializer 权重初始化的方法。 不同的层可能使用不同的关键字来传递初始化方法,一般来说指定初始化方法的关键字是kernel_initializer 和 bias_initializer model.add(Den 阅读全文
posted @ 2022-08-06 00:31 理想几岁 阅读(730) 评论(0) 推荐(0) 编辑