fix multigpu extractor

2019-10-10 13:01:57 +04:00
parent 9b08199e87
commit d2d0dcb5e7
2 changed files with 2 additions and 2 deletions
@@ -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