pointnet train函數第二十六句 ops

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。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章