Open FatPandao opened 3 years ago
In your code
# Expand temperature to match the size of logits temperature = self.temperature.unsqueeze(1).expand(logits.size(0), logits.size(1)) return logits / temperature
Why not just doing something like
return logits / self.temperature
In your code
Why not just doing something like