chanind / frame-semantic-transformer

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

Add ability to set which gpu to use #23

Closed striebel closed 1 year ago

striebel commented 1 year ago

Updated handling of the use_gpu parameter of the FrameSemanticTransformer object constructor, so in addition to being able to pass True or False, alternatively the index of a specific gpu to use can be passed. This is helpful when multitasking.