search

chanind / frame-semantic-transformer

Frame Semantic Parser based on T5 and FrameNet
https://chanind.github.io/frame-semantic-transformer
MIT License
51 stars 10 forks source link

GPU Acceleration Available? #7

Closed RiverDong closed 1 year ago

RiverDong commented 1 year ago

Hi I am wondering if we can use GPU to accelerate the inference time? Right now it's somewhat slow using CPU. Thank you

chanind commented 1 year ago

It should default to using GPU if available. How are you running this currently?

chanind commented 1 year ago

Closing for now. Please re-open the issue if you're still having trouble using the GPU.