diff options
Diffstat (limited to 'models/fpfe.py')
| -rw-r--r-- | models/fpfe.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/models/fpfe.py b/models/fpfe.py index 29f7a5f..28a0440 100644 --- a/models/fpfe.py +++ b/models/fpfe.py @@ -1,5 +1,4 @@ -from torch import nn as nn -from torch.nn import functional as F +import torch.nn as nn from models.layers import FocalConv2d |