tensoflow
文章平均质量分 94
紫薯真好吃
制定计划,不拖沓并且持之以恒~
展开
-
keras 保存模型训练过程中的checkpoint
from __future__ import print_functionimport kerasfrom keras.datasets import cifar10from keras.preprocessing.image import ImageDataGeneratorfrom keras.models import Sequentialfrom keras.layers im...原创 2020-04-08 14:18:14 · 1238 阅读 · 0 评论 -
tensorflow保存模型,保存训练过程中的checkpoint
1、使用tf.keras.callbacks.ModelCheckpoint()定义回调函数tf.keras.callbacks.ModelCheckpoint(),在model.fit()中加入该回调函数,将在model训练时自动调用回调函数保存训练过程记录和模型结构以及参数权重,这些内容将保存在一个.ckpt文件中。、示例:import osimport tensorflow ...原创 2020-04-08 14:04:29 · 5724 阅读 · 0 评论 -
【keras-bert 学习笔记】2. 保存、加载预训练模型,在预训练模型上添加层做监督训练(fine tune)
1.预训练模型,并保存import osimport tensorflow as tffrom keras_bert import (get_model, compile_model, get_base_dict, gen_batch_inputs)from indoor_location.utils import get_sentence_pairsseqence_len = ...原创 2020-04-08 13:29:35 · 3426 阅读 · 2 评论 -
TF:tf.losses.sparse_softmax_cross_entropy
tf.losses.sparse_softmax_cross_entropy(logits=logits, labels=labels)等价与tf.nn.sparse_softmax_cross_entropy_with_logits(logits=logits, labels=labels) 加 tf.reduce_mean。它们的参数维度:logits: [batch_size, ...转载 2020-04-06 17:41:05 · 1053 阅读 · 0 评论 -
anaconda 安装tensorflow 2.1
1.Anaconda python3.7版本安装2.Tensorflow 2.1安装3.PyCharm安装4.Jupyter Notebook安装Anaconda 安装前往Anaconda官网Python 3.7 版本,选择64-Bit Graphical Installer下载。下载完成后按照引导安装,采用默认路径。在Advanced Installation Option...转载 2020-04-06 17:35:32 · 3697 阅读 · 2 评论 -
TensorFlow tf.keras.losses.SparseCategoricalCrossentropy 、 categorical_crossentropy
categorical_crossentropy VS. sparse_categorical_crossentropy转载qq_42961707最后发布于2019-07-13 22:10:12阅读数 5022收藏categorical_crossentropy 和 sparse_categorical_crossentropy 的区别在哪?如果你的 targets 是 one...转载 2020-04-06 16:07:27 · 2091 阅读 · 0 评论 -
keras序列化模型 to json文件,保存模型和加载模型
首先保存1、模型结构,2、模型参数数据#!/usr/bin/env python# -*- coding:utf-8 -*-# Author: Jia ShiLin'''#模型的权重保存在HDF5中#模型的结构保存在JSON文件或者YAML文件中''''''#Keras提供了to_json()生成模型JSON描述,并将模型的JSON描述保存到文件中,#反...转载 2020-04-06 12:27:02 · 1257 阅读 · 0 评论 -
加载keras内置dataset在线下载出错——使用下载好的数据集
1、原因分析:keras中cifar10.py模块的load_data(),其中的path = get_file(dirname, origin=origin, untar=True)将从默认的 C:\Users\sunxu\.keras\datasets路径下返回数据集路径,如果该路径下有cifar-10-batches-py.tar.gz文件则不用从网上在线下载,否则将会联网下载(可能下...原创 2020-04-05 12:09:25 · 2760 阅读 · 0 评论 -
tensorflow中用saver保存模型
我们在用tensorflow训练模型时,可能需要训练很长很长一段时间,为了方便下次使用,应该将模型保存起来。在sklearn中,我们可以使用pickle模块进行模型保存;而在tensorflow中,我们可以使用它自带的Saver()类进行模型的保存。(一)Saver类 Saver类是用于保存和恢复变量的。它有将变量保存到checkpoint和从checkpoint中恢复变量...转载 2020-04-03 11:28:51 · 860 阅读 · 0 评论 -
tf.keras模块——Input、Model
tf.keras.Input() 初始化一个keras张量 1 2 3 4 5 6 7 8 9 tf.keras.Input( shape=None, batch_size=None, name=None, dtype=None,...转载 2020-04-03 10:13:30 · 5840 阅读 · 0 评论 -
tensorflow使用GPU训练
1. 查看gpu的使用详情:(1)查看gpu使用情况nvidia-smi.exe # windows上nvidia-smi -l #linux服务器上# 显示的结果中 Volatile GPU-Util:浮动的GPU利用率;(2)linux上查看进程占用gpu的情况:##实用技巧:##如果你在linux终端运行深度学习python脚本文件,运行中发现占用多个GPU和内存...原创 2020-03-04 22:07:29 · 10099 阅读 · 0 评论 -
tf.train.Saver()-tensorflow中模型的保存及读取
tf.train.Saver()-tensorflow中模型的保存及读取转自:https://www.cnblogs.com/bevishe/p/10359993.html作用:训练网络之后保存训练好的模型,以及在程序中读取已保存好的模型使用步骤:实例化一个Saver对象saver = tf.train.Saver() 在训练过程中,定期调用saver.save方法,像文件夹...转载 2020-02-26 15:08:00 · 763 阅读 · 0 评论