We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e9c321e commit a1735efCopy full SHA for a1735ef
README.md
@@ -44,7 +44,7 @@ def dice_loss(x1, x2):
44
union = torch.sum(x1 + x2, dim=dim)
45
return 1 - (2. * inter / union).mean()
46
47
-reg = AffineRegistration(dissimilairity_function=dice_loss, optimizer=torch.optim.Adam)
+reg = AffineRegistration(dissimilarity_function=dice_loss, optimizer=torch.optim.Adam)
48
```
49
CUDA support (NVIDIA GPU)
50
```python
0 commit comments