Add simple baseline for LFNA

This commit is contained in:
D-X-Y
2021-04-29 16:30:47 +08:00
parent 2c56938ee7
commit 14905d0011
8 changed files with 296 additions and 307 deletions

View File

@@ -18,11 +18,3 @@ def obtain_accuracy(output, target, topk=(1,)):
correct_k = correct[:k].view(-1).float().sum(0, keepdim=True)
res.append(correct_k.mul_(100.0 / batch_size))
return res
class EvaluationMetric(abc.ABC):
def __init__(self):
self._total_metrics = 0
def __len__(self):
return self._total_metrics