- 博客(11)
- 资源 (1)
- 收藏
- 关注
原创 translation attention
attention前言将keras数据集下载到本地 datasets dirout二、使用步骤random_stateattention weights总结前言将keras数据集下载到本地 datasets dir # Download the file path_to_zip = tf.keras.utils.get_file( 'spa-eng.zip', origin='http://storage.googleapis.com/download.tensorfl
2020-12-25 10:06:45 190
原创 注意力 image caption
bert fine tuningtf2 checkpoints berttf2 checkpoints bert对应:gs_folder_bert = "gs://cloud-tpu-checkpoints/bert/keras_bert/uncased_L-12_H-768_A-12"tf.io.gfile.listdir(gs_folder_bert)tf2 checkpoints bert
2020-12-24 11:21:25 254
原创 终极解决显存分配
项目场景:TF Version: 2.4.0-dev20201015TF-Hub version: 0.9.0Eager mode enabled: True问题描述:InternalError: Blas SGEMM launch failed解决方案:gpus = tf.config.experimental.list_physical_devices(device_type='GPU') for gpu in gpus: tf.config.experime
2020-12-17 19:08:35 865
原创 深度学习代码积累
文章目录为方便打印调试打印9宫格如果最后不用激活函数 logitnext iter 只迭代一个batch数据集自定义模型保存与恢复模型保存训练结果为方便打印调试记录行号和文件名 变量值和类型 def print_variables(var, line_no): import inspect frame = inspect.currentframe() # __FILE__ fileName = frame.f_code.co_fil
2020-12-15 02:45:50 930 1
原创 保存加载模型model.save()
当savemodel hd5 时需要 metricstf.metrics.SparseCategoricalAcc uracy() 不能是 accuracy 字符串 否则when load model 测试精确度会有问题。将产生怀疑 def create_model(): model = tf.keras.models.Sequential([ keras.layers.Dense(512, activation='relu', input_shape=
2020-12-15 02:41:49 15684
原创 take skip 划分训练集
文章目录加载npz在文件绘制精度和损失保存训练过程把compile和fit放在一个函数里断点训练模型使用take和skip划分验证集总结加载npz在文件绘制精度和损失def plot_cifar_loss(npz_path): """ read flower.npz and plot acc val_acc loss val_loss ... """ data = np.load(npz_path) # print(data.files) acc
2020-12-13 21:17:19 504 2
原创 tensorflow2代码积累
深度学习代码模型代码深度学习代码保存与加载npz文件plot绘制一行两列子图精度和损失函数式模型序列式模型绘制数据增强图片调用预处理层放缩图片到(0-255)->(0-1)函数式模型keras加载测试数据集验证测试集evaluate保存与加载npz文件数组形式 acc = history.history['accuracy'] val_acc = history.history['val_accuracy'] loss = history.history['loss']
2020-12-12 20:00:49 263
原创 tensorflow2 代码积累
tensorflow2按需申请显存rtx3080兼容tf1按需申请显存tf2 按需申请显存保存模型加载保存的模型保存checkpoint 断点继续训练继续训练预测rtx3080兼容tf1按需申请显存代码主要参考了简单粗暴tensorflow2我的tensorflow版本是:tf-nightly-gpu 2.4.0.dev20201015import tensorflow.compat.v1 as tftf.disable_eager_execution()config =
2020-12-11 21:15:51 297
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人