File tree Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Original file line number Diff line number Diff line change 3
3
from .conv import Conv1d , Conv2d , Conv3d # noqa: F401
4
4
from .convert import continual , forward_stepping # noqa: F401
5
5
from .delay import Delay # noqa: F401
6
- from .interface import CoModule , TensorPlaceholder , FillMode # noqa: F401
6
+ from .interface import CoModule , FillMode , TensorPlaceholder # noqa: F401
7
7
from .pooling import ( # noqa: F401
8
8
AdaptiveAvgPool2d ,
9
9
AdaptiveAvgPool3d ,
Original file line number Diff line number Diff line change 1
1
""" Register modules with conversion system and 3rd-party libraries """
2
2
3
3
from functools import wraps
4
- from typing import Type , Callable
5
- from torch import nn , Tensor
4
+ from typing import Callable , Type
5
+
6
+ from torch import Tensor , nn
7
+
6
8
from .batchnorm import BatchNorm2d
7
9
from .container import Sequential
8
10
from .conv import Conv1d , Conv2d , Conv3d
You can’t perform that action at this time.
0 commit comments