added a clear statement of device to use

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

Closed krzysztoffiok closed 3 years ago

krzysztoffiok commented 3 years ago

This fix makes the user state clearly which device he or she wishes to use in the experiment.