refactor: remove if-condition

3 years ago · 56a6b38f76
parent 499c4749d7
commit 56a6b38f76
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()