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

Remove checking `smoothing` parameter
2 years ago · a517bf6a7a
parent 447677616f 56a6b38f76
commit a517bf6a7a
1 changed files with 1 additions and 4 deletions
--- a/benchmark.py
+++ b/benchmark.py
@ -310,10 +310,7 @@ 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.loss = nn.CrossEntropyLoss().to(self.device)
        self.target_shape = tuple()

        self.optimizer = create_optimizer_v2(