nerf-pytorch/conftest.py at bcb670095eefaf798d1cdde4592f2658a6b31085 - nerf-pytorch - Gitea: Service Git d'Aurore

Suppression de la possibilité de cloner des dépôts privés en HTTPS ;
Réinstallation de drone ou changement d'outil de CI ;
Possiblement migration de Gitea à Forgejo.

You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

12 lines

251 B

Python

Raw Blame History

 import pytest
 import torch
 devices = {'cpu': torch.device('cpu')}
 if torch.cuda.is_available():
     devices['cuda'] = torch.device('cuda:0')
 @pytest.fixture(params=devices.values(), ids=devices.keys())
 def device(request):
     return request.param