Tensorflow
HighDS
这个作者很懒,什么都没留下…
展开
-
keras 利用ModelCheckpoint实现断点续训
一个回归预测的问题,实现断开模型后还能从上次的权重继续训练。这里利用ModelCheckpoint实现保存模型。nn_model = Sequential()nn_model.add(Dense(32, input_dim=4, activation='relu'))nn_model.add(Dense(16, input_dim=32, activation='relu'))nn_mod...原创 2020-02-06 02:20:27 · 1696 阅读 · 0 评论 -
window10 如何安装GPU版本的tensorflow, 正确安装cuda, cudnn
window10 系统如何安装GPU版本的tensorflow, 并安装正确的cuda, cudnn第一步: 查看显卡是否支持cuda(如果不支持,爱莫能助,装cpu版本吧)https://developer.nvidia.com/cuda-gpus是否支持某个版本的cuda,还得参考显卡驱动的要求。(建议直接粗暴安装,安装失败再升级显卡驱动)https://docs.nvidia.com...原创 2020-01-17 01:34:33 · 601 阅读 · 0 评论