Update for compatibility with Python 3.12 and Radare 6.0.2
Tested with the following workflows: vtable diff deep trace -> vtable alignment
This commit is contained in:
+3
-1
@@ -326,7 +326,9 @@ def load_model(path, device="cpu"):
|
||||
|
||||
|
||||
def accuracy(y, probs):
|
||||
return torch.mean(torch.tensor([torch.sum(probs[i][yi]) for i, yi in enumerate(y)]))
|
||||
return torch.mean(
|
||||
torch.tensor([torch.sum(probs[i][yi].detach()) for i, yi in enumerate(y)])
|
||||
)
|
||||
|
||||
|
||||
def cosine_similarities(model):
|
||||
|
||||
Reference in New Issue
Block a user