Add missing import

1713175349 · May 14, 2021 · 96501a2 · 96501a2
1 parent 62d1bcb
commit 96501a2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchmdnet/models/utils.py b/torchmdnet/models/utils.py
@@ -2,6 +2,7 @@
 import math
 import torch
 from torch import nn
+import torch.nn.functional as F
 from torch_scatter import scatter
 from torch_geometric.nn import MessagePassing