search

robinvanschaik / interpret-flair

A small repository to test Captum Explainable AI with a trained Flair transformers-based text classifier.
MIT License
24 stars 2 forks source link

added the fix for using proper flair.device #6

Closed krzysztoffiok closed 3 years ago

krzysztoffiok commented 3 years ago

This is the device fix we were discussing in https://github.com/robinvanschaik/interpret-flair/issues/5