Output node name? - Githubissues

nyu-dl / dl4marco-bert

BSD 3-Clause "New" or "Revised" License

476 stars 87 forks source link

Output node name? #4

Closed bradleymackey closed 5 years ago

bradleymackey commented 5 years ago

Hi,

Thanks for the great repo. I'm trying to do a conversion on your pre-trained model to PyTorch (via MMdnn), but I'm unable to figure out what the output node name of the model is. Would you happen to know what this is?

Thanks, Bradley

bradleymackey commented 5 years ago

No longer needed. There is a BERT conversion script available in the pytorch-pretrained-bert repository. Thanks again!