Update model_dtat.py

model_dtat.py

@@ -317,7 +317,7 @@ class DTATTransformer(nn.Module):
             logits = logits.view(B*T, C)
             targets = targets.view(B*T)
             # Calculate loss directly in BPC instead of nats
-            loss = F.cross_entropy(logits, targets) * math.log2(math.e)
+            loss = F.cross_entropy(logits, targets) / math.log(2)
         
         return logits, loss, importance_scores