--- title: modeling.text2text.core keywords: fastai sidebar: home_sidebar summary: "This module contains core custom models, loss functions, etc... for text2text based tasks (e.g., language modeling, summarization, translation, etc...)" description: "This module contains core custom models, loss functions, etc... for text2text based tasks (e.g., language modeling, summarization, translation, etc...)" nb_path: "nbs/02za_modeling-text2text-core.ipynb" ---
{% raw %}
{% endraw %} {% raw %}
{% endraw %} {% raw %}
torch.cuda.set_device(1)
print(f'Using GPU #{torch.cuda.current_device()}: {torch.cuda.get_device_name()}')
Using GPU #1: GeForce GTX 1080 Ti
{% endraw %}

Cleanup