Skip to content

Commit 07f89a2

Browse files
authored
Update metrics.py
Added white space before and after sigmoid_np
1 parent e4a55ab commit 07f89a2

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

algoperf/workloads/ogbg/metrics.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,10 @@ def compute(self):
4949
dist.all_gather(all_tensors, tensor)
5050
all_values[idx] = torch.cat(all_tensors).cpu().numpy()
5151
labels, logits, mask = all_values
52+
5253
def sigmoid_np(x):
5354
return 1 / (1 + np.exp(-x))
55+
5456
sigmoid = sigmoid_np
5557

5658
mask = mask.astype(bool)

0 commit comments

Comments
 (0)