You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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:
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!
The text was updated successfully, but these errors were encountered:
Hello, I believe I found a minor bug in
IntentAccuracyDailyDialog
, lines 672-3 inenvs/text_generation/metric.py
. The device is currently set with the following two lines: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!The text was updated successfully, but these errors were encountered: