--- title: modeling.language_modeling keywords: fastai sidebar: home_sidebar summary: "This module contains custom models, loss functions, custom splitters, etc... for both causal and MLM language modeling tasks" description: "This module contains custom models, loss functions, custom splitters, etc... for both causal and MLM language modeling tasks" nb_path: "nbs/02a_modeling-language-modeling.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