关于tensorflow迁移学习那些事

首先在看本内容之前,推荐你去看tensorflow官网的迁移学习和微调,那个讲的更清楚,也更加详细,网址如下所示。但是官网里面的模型加载确实在网络上下载的,不能符合实际的需求,因为有些情况,我们想用自己预训练的模型加载并且进行微调,那么就是本内容的重点。https://tensorflow.google.cn/guide/keras/transfer_learning?hl=zh_cn

大致的内容步骤如下

1、保存自己预训练的模型

2、加载自己的模型

3、进行微调

那么首先第一步,保存自己与训练的模型

我们首先应该调用model.save_weights()这个函数,保存自己的模型。数据集自自己调整下。

import tensorflow as tf
import numpy as np
from tensorflow.keras.layers import Conv1D,BatchNormalization,Dense,Activation,GlobalAveragePooling1D
from tensorflow.keras.regularizers import l2


auto_train_x_a = np.load("U_process_train_x.npy")[0:30,:]
auto_train_y = np.load("auto_process_data_train_y.npy")[0:30,:]

def auto():
    x_input = tf.keras.Input(shape=(128,))
    x1 = tf.keras.layers.Dense(128,kernel_regularizer=tf.keras.regularizers.l2(1e-4))(x_input)
    x2 = tf.keras.layers.Dense(32
  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值