diff options
| -rw-r--r-- | models/auto_encoder.py | 4 | ||||
| -rw-r--r-- | models/fpfe.py | 3 | ||||
| -rw-r--r-- | models/tfa.py | 2 |
3 files changed, 4 insertions, 5 deletions
diff --git a/models/auto_encoder.py b/models/auto_encoder.py index 6483bd9..4a8aaa5 100644 --- a/models/auto_encoder.py +++ b/models/auto_encoder.py @@ -1,6 +1,6 @@ import torch -from torch import nn as nn -from torch.nn import functional as F +import torch.nn as nn +import torch.nn.functional as F from models.layers import VGGConv2d, DCGANConvTranspose2d 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 diff --git a/models/tfa.py b/models/tfa.py index 72bfcfb..b80328a 100644 --- a/models/tfa.py +++ b/models/tfa.py @@ -1,7 +1,7 @@ import copy import torch -from torch import nn as nn +import torch.nn as nn from models.layers import BasicConv1d |