Merge pull request #1105 from kozistr/refactor/remove-condition

Remove checking `smoothing` parameter
3 years ago · a517bf6a7a
parent 447677616f 56a6b38f76
commit a517bf6a7a
1 changed files with 1 additions and 4 deletions
--- a/benchmark.py
+++ b/benchmark.py
@ -310,9 +310,6 @@ class TrainBenchmarkRunner(BenchmarkRunner):
        super().__init__(model_name=model_name, device=device, torchscript=torchscript, **kwargs)
        self.model.train()

-        if kwargs.pop('smoothing', 0) > 0:
-            self.loss = nn.CrossEntropyLoss().to(self.device)
-        else:
        self.loss = nn.CrossEntropyLoss().to(self.device)
        self.target_shape = tuple()