refactor: remove if-condition

pull/1105/head
kozistr 3 years ago
parent 499c4749d7
commit 56a6b38f76

@ -310,10 +310,7 @@ class TrainBenchmarkRunner(BenchmarkRunner):
super().__init__(model_name=model_name, device=device, torchscript=torchscript, **kwargs) super().__init__(model_name=model_name, device=device, torchscript=torchscript, **kwargs)
self.model.train() self.model.train()
if kwargs.pop('smoothing', 0) > 0: self.loss = nn.CrossEntropyLoss().to(self.device)
self.loss = nn.CrossEntropyLoss().to(self.device)
else:
self.loss = nn.CrossEntropyLoss().to(self.device)
self.target_shape = tuple() self.target_shape = tuple()
self.optimizer = create_optimizer_v2( self.optimizer = create_optimizer_v2(

Loading…
Cancel
Save