Tensorflow2.x的记录

一、keras与eager执行模式

eager执行模式可以仅仅通过组装层就可以定义、训练并评估模型

二、Eager模式

在tf2.x中,默认执行的是eager模式
1.eager模式返回的是tf.Tensorf对象,而静态图模式执行回话后,返回的是Numpy数组对象
2.在eager模式中,print返回的tf.Tensor对象与静态图的区别:
静态图返回例子的是:

1. tf.GradientTape()自动微分运算

2. 自定义训练循环train()

定制train_step函数用在训练循环中:

def train(num_classes, batch_size):
    """
    定制 训练循环
    :param num_classes:
    :return:
    """
    # 加载模型
    net = AlexNet(num_classes)
    net.build((4,224,224,3))
    # 构建数据集
    dataset = make_dataset(10, 5, 4)
    # 损失函数
    # loss = tf.losses.CategoricalCrossentropy(from_logits=False)     #标签是one-hot形式
    loss = tf.losses.SparseCategoricalCrossentropy(from_logits=False) #标签不是one-hot形式
    # 优化器
    optimizer = tf.optimizers.Adam()

    # 配置
    step = tf.Variable(1, dtype=tf.int32, name="global_step")
    accuracy = tf.metrics.Accuracy()

    mean_loss = tf.metric
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值