diff options
Diffstat (limited to 'utils/triplet_loss.py')
| -rw-r--r-- | utils/triplet_loss.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/utils/triplet_loss.py b/utils/triplet_loss.py index c3e5802..22ac2ab 100644 --- a/utils/triplet_loss.py +++ b/utils/triplet_loss.py @@ -1,4 +1,4 @@ -from typing import Optional +from typing import Optional, Tuple import torch import torch.nn as nn @@ -79,7 +79,7 @@ class JointBatchTripletLoss(BatchTripletLoss): self, hpm_num_parts: int, is_hard: bool = True, - margins: tuple[float, float] = (0.2, 0.2) + margins: Tuple[float, float] = (0.2, 0.2) ): super().__init__(is_hard) self.hpm_num_parts = hpm_num_parts |