search

timoschick / pet

This repository contains the code for "Exploiting Cloze Questions for Few-Shot Text Classification and Natural Language Inference"
https://arxiv.org/abs/2001.07676
Apache License 2.0
1.62k stars 283 forks source link

TensorFlow 2.4 support is not available #22

Closed Rishabbh-Sahu closed 3 years ago

Rishabbh-Sahu commented 3 years ago

Hi Timo Schick,

Thanks for creating this repo. I've been looking to implement this using TensorFlow 2.4 and multi-gpu setup for dialog-nlu task. Are you planning to release Tensorflow 2.4 version for the same which can be converted to tflite to test in mobile devices.

Thanks.

timoschick commented 3 years ago

Currently I do not have plans to release a version of PET that is compatible with TensorFlow 2.4. If you want to implement PET yourself in a way that is compatible with TensorFlow, let me know if you have any specific questions.

Rishabbh-Sahu commented 3 years ago

Thanks for the reply. Will surely get in touch with you.

danielkorat commented 2 years ago

Hi @Rishabbh-Sahu Did you implement/find a TensorFlow version? Thanks :)