From fbbc6a2dc533c480059dd8b621ff5abef349b593 Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Thu, 24 Dec 2020 18:51:33 +0800 Subject: Optimize imports --- models/auto_encoder.py | 4 ++-- models/fpfe.py | 3 +-- models/tfa.py | 2 +- 3 files changed, 4 insertions(+), 5 deletions(-) (limited to 'models') 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 -- cgit v1.2.3