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

被折叠的 条评论
为什么被折叠?



