phate334 commited on
Commit
6bedbda
1 Parent(s): 6eed107

[add] show gpu flops when choice device.

Browse files
Files changed (1) hide show
  1. main.py +1 -1
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