pointnet train函数第二十六句 ops

原创于 2020-06-28 21:28:40 发布 · 273 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow

pointnet 专栏收录该内容

40 篇文章

订阅专栏

本文详细解析了在TensorFlow中如何构建训练流程，通过将模型输入、输出、预测、损失等关键元素组织成操作（op），并存储在一个字典中，以便在训练循环中高效调用。这种方式简化了复杂模型的训练过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

ops = {'pointclouds_pl': pointclouds_pl,
               'labels_pl': labels_pl,
               'is_training_pl': is_training_pl,
               'pred': pred,
               'loss': loss,
               'train_op': train_op,
               'merged': merged,
               'step': batch}

ops是一个dict，将上文构建的输入，label，is_training_pl，pred，loss等放到一个dict里面，共后面进行训练的时候将这些参数传递进实际训练loop中

在tensorflow中，一个tensor可以当作一个op也可以多个tensor组成一个op。这句代码就是把前面的模型以及输入输出都作为op，放在一个dict里面，这样后续的train one epoch里面通过ops里面的名字找到对应的op然后在session.run中来计算这些op。