diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-24 18:51:33 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-24 18:56:23 +0800 |
| commit | fbbc6a2dc533c480059dd8b621ff5abef349b593 (patch) | |
| tree | ad8eb3c31b7445388bdcfbd6f443bfb34eeb4600 /models | |
| parent | 4ad6d0af2f4e6868576582e4b86b096a2e95e6fa (diff) | |
Optimize imports
Diffstat (limited to 'models')
| -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 |