深度学习基础--logits

logits

  深度学习源码里经常出现logits
  logits: 未归一化的概率, 一般也就是 softmax层的输入。所以logits和lables的shape一样

  • 3
    点赞
  • 13
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
迁移学习是指在一个任务上训练好的模型可以被应用到另外一个任务上,它可以大大减少模型训练的时间和资源消耗,同时也提高了模型的泛化能力。下面是一个基于 TensorFlow 的迁移学习模型代码示例: ```python import tensorflow as tf import numpy as np from tensorflow import keras from tensorflow.keras import layers # 加载预训练模型 base_model = keras.applications.MobileNetV2( weights='imagenet', # 使用 ImageNet 数据集预训练的权重 input_shape=(224, 224, 3), include_top=False) # 去掉顶层全连接层 # 冻结模型底层 base_model.trainable = False # 添加新的全连接层 inputs = keras.Input(shape=(224, 224, 3)) x = base_model(inputs, training=False) x = keras.layers.GlobalAveragePooling2D()(x) x = keras.layers.Dense(1024, activation='relu')(x) outputs = keras.layers.Dense(10, activation='softmax')(x) # 构建新模型 model = keras.Model(inputs, outputs) # 编译模型 model.compile(optimizer=keras.optimizers.Adam(), loss=keras.losses.SparseCategoricalCrossentropy(from_logits=True), metrics=[keras.metrics.SparseCategoricalAccuracy()]) # 加载新的数据集 (x_train, y_train), (x_test, y_test) = keras.datasets.cifar10.load_data() x_train = x_train.astype('float32') / 255. x_test = x_test.astype('float32') / 255. # 训练模型 model.fit(x_train, y_train, batch_size=32, epochs=10, validation_data=(x_test, y_test)) ``` 在以上示例中,我们使用了 MobileNetV2 模型作为迁移学习的基础模型,并且加载了 ImageNet 数据集预训练的权重。我们去掉了顶层全连接层,然后添加了新的全连接层和输出层,构建了一个新的模型。在训练时,我们使用了 CIFAR-10 数据集,并且只训练了新加的全连接层和输出层。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值