youtube-8m
youtube-8m copied to clipboard
Enable training with segment_labels.
Right now, the segment_labels flag in train.py is not fully functional.
Hi Zheng, I managed to do training with segment_labels only by adding a lookup table. The goal is to be able to train both the video models and segment models with segment labels only. Can do a PR if it helps.
@elv-youliangyu Thanks! Feel free to start a PR and I will be happy to review it. :) :+1:
Hi all, can you tell me the place to add lookup table in train.py?