search

yuvalkirstain / s2e-coref

MIT License
45 stars 15 forks source link

predict.py script for any input documents #3

Closed ariecattan closed 2 years ago

ariecattan commented 3 years ago

Great work!

Do you plan to make a predict.py script to run your model on custom input documents (like in https://github.com/lxucs/coref-hoi and https://github.com/mandarjoshi90/coref)

This will be really helpful because these models require a lot of GPU memory

Thanks!

yuvalkirstain commented 2 years ago

Hey! We do not intend to create such a script. But if you would like to create one, we would love to merge it into our repo and make it more usable.