Преглед изворни кода

remove cbrt which doesnt exist on python 3.9

Alex Cheema пре 10 месеци
родитељ
комит
4a7503551a
1 измењених фајлова са 1 додато и 1 уклоњено
  1. 1 1
      exo/viz/topology_viz.py

+ 1 - 1
exo/viz/topology_viz.py

@@ -161,7 +161,7 @@ class TopologyViz:
 
     # Calculate total FLOPS and position on the bar
     total_flops = sum(self.topology.nodes.get(partition.node_id, UNKNOWN_DEVICE_CAPABILITIES).flops.fp16 for partition in self.partitions)
-    bar_pos = (math.tanh(math.cbrt(total_flops)/2.5 - 2) + 1)
+    bar_pos = (math.tanh(total_flops**(1/3)/2.5 - 2) + 1)
 
     # Add GPU poor/rich bar
     bar_width = 30