fix multigpu extractor

This commit is contained in:
Colombo committed 2019-10-10 13:01:57 +04:00
1 parent 9b08199e87
commit d2d0dcb5e7
2 files changed
+2 -2

No files matched your search

+1 -1
View File
@@ -102,7 +102,7 @@ class device:
elif device.backend == "tensorflow":
for dev in cuda_devices:
if dev['total_mem'] >= totalmemsize_gb*1024*1024*1024:
result.append (i)
result.append ( dev['index'] )
return result