diff options
Diffstat (limited to 'models/auto_encoder.py')
| -rw-r--r-- | models/auto_encoder.py | 4 |
1 files changed, 2 insertions, 2 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 |