📝Transformers

• tags: 🏷Python
• up: 📂データサイエンス
• refs.
  • 🔗Transformers

🔧TransformersでGPUを有効にする方法 #

このおまじないが必要.

import torch
device = "cuda:0" if torch.cuda.is_available() else "cpu"

References #

• 🖊AIによる自動エロ文章生成! - Transformers をGoogle Cloud Runにデプロイしてオンライン文章要約 | Futurismo