IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
Pārlūkot izejas kodu

Small fix.

master
Stanislaw Adaszewski pirms 4 gadiem
vecāks
revīzija
915c568490
1 mainītis faili ar 3 papildinājumiem un 0 dzēšanām
  1. +3
    -0
      src/icosagon/data.py

+ 3
- 0
src/icosagon/data.py Parādīt failu

@@ -31,6 +31,9 @@ def _equal(x: torch.Tensor, y: torch.Tensor):
indices_y = list(map(tuple, y.indices().transpose(0, 1)))
order_y = sorted(range(len(indices_y)), key=lambda idx: indices_y[idx])
if not indices_x == indices_y:
return torch.tensor(0, dtype=torch.uint8)
return (x.values()[order_x] == y.values()[order_y])


Notiek ielāde…
Atcelt
Saglabāt