allenai / allenai/RL4LMs

CPU Support Minor Bug

Abierto
#59 0 comentarios 0 reacciones 0 asignados Ver en GitHub
Lenguaje dominante
Python
Estrellas
2.4k
Forks
201
Métricas de merge de PR
Sin PR fusionados en 30 d

Descripción

Hello, I believe I found a minor bug in `IntentAccuracyDailyDialog`, lines 672-3 in `envs/text_generation/metric.py`. The device is currently set with the following two lines:
```
self._device = "cuda" if torch.cuda.is_available() else "cpu"
self._device = f"cuda:{torch.cuda.device_count() - 1}"
```
However, I believe it should be such that the device is `f"cuda:{torch.cuda.device_count() - 1}"` if on GPU and `"cpu"` otherwise (as currently it can never be set to CPU, and instead tries to find `"cuda:-1"`). Thanks so much for putting this library together!

Guía de contribución

No hay ninguna guía de contribución indexada para este repositorio

Evaluación

Este issue todavía no se ha evaluado.

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.