tensorflow
wshzd
机器学习,深度学习,NLP,强化学习
展开
-
ckpt文件restore的两种方法
这里以BERT恢复模型为例,说明恢复ckpt的两种常见方法# 方法一:init_checkpoint = "chinese_L-12_H-768_A-12/bert_model.ckpt"use_tpu = False# 获取模型中所有的训练参数。tvars = tf.trainable_variables()# 加载BERT模型(assignment_map, initia...原创 2019-04-28 16:15:57 · 2247 阅读 · 0 评论 -
使用胶囊网络实现MNIST数据分类
#!/usr/bin/python# -*- coding:utf-8 -*-# https://www.sohu.com/a/226611009_633698import numpy as npimport osimport tensorflow as tfimport matplotlib.pyplot as plt# Reset the default graph fo...转载 2019-05-08 19:12:40 · 2813 阅读 · 1 评论 -
tensorflow2.0实现MNIST分类
首先需要安装tensorflow2.0版本,这个版本与1.x的版本变化比较大,主要采用了keras的API,具体可以参考一、下面是以jupyter notebook开发环境为例进行代码演示,jupyter notebook需要在命令行签名增加!!pip install tensorflow-gpu==2.0.0-alpha0二、导入使用到的库from __future__ impo...原创 2019-05-13 20:26:00 · 3672 阅读 · 1 评论 -
tensorflow2.0实现MNIST分类(二)
下面是tensorflow2.0完全采用keras的API来实现的from __future__ import absolute_import, division, print_function, unicode_literals!pip install -q tensorflow==2.0.0-alpha0import tensorflow as tfmnist = t...原创 2019-05-13 20:29:43 · 7019 阅读 · 0 评论 -
tensorflow保存模型为pd+variable格式
import tensorflow as tffrom tensorflow import saved_model as smX = tf.placeholder(tf.float32, shape=(3,))scale = tf.Variable([10, 11, 12], dtype=tf.float32)y = tf.multiply(X, scale)with tf.Ses...原创 2019-06-17 11:34:38 · 1083 阅读 · 0 评论 -
tensorflow模型压缩
通常训练的模型存储为float32,如果可以转换为float16来进行存储的话,模型大小会减少一半把已经训练好的模型使用TensorFlow Lite converter进行转换为float16即可importtensorflowastfconverter=tf.lite.TFLiteConverter.from_saved_model(saved_model_dir)con...原创 2019-08-28 18:44:46 · 905 阅读 · 0 评论 -
autokeras报错
import autokeras报错信息如下:ModuleNotFoundError: No module named 'kerastuner.tuners'解决方案如下:git clonehttps://github.com/keras-team/keras-tunercd keras-tunerpip install ....原创 2019-08-29 17:13:53 · 963 阅读 · 1 评论