Spaces:
Running
Running
[add] show gpu flops when choice device.
Browse files
main.py
CHANGED
@@ -19,7 +19,7 @@ with open("devices.json", "r", encoding="utf-8") as f:
|
|
19 |
devices = {key: Device(**value) for key, value in data.items()}
|
20 |
|
21 |
device_options = [
|
22 |
-
f"{key} (Memory: {value.memory_size}GB, Bandwidth: {value.memory_bandwidth}GB/s)"
|
23 |
for key, value in devices.items()
|
24 |
]
|
25 |
|
|
|
19 |
devices = {key: Device(**value) for key, value in data.items()}
|
20 |
|
21 |
device_options = [
|
22 |
+
f"{key} (Memory: {value.memory_size}GB, FLOPS: {value.FLOPS}, Bandwidth: {value.memory_bandwidth}GB/s)"
|
23 |
for key, value in devices.items()
|
24 |
]
|
25 |
|