Skip to content
Snippets Groups Projects
Commit e46a1211 authored by Daniel CARRON's avatar Daniel CARRON :b: Committed by André Anjos
Browse files

Updated typing imports im models

parent d5c0f038
Branches
Tags
1 merge request!6Making use of LightningDataModule and simplification of data loading
Hide whitespace changes
Inline Side-by-side
src/ptbench/models/alexnet.py
+ 2
− 1
View file @ e46a1211
......@@ -13,7 +13,8 @@ import torch.utils.data
import torchvision.models as models
import torchvision.transforms
from ..data.typing import Checkpoint, DataLoader, TransformSequence
from ..data.typing import DataLoader, TransformSequence
from .typing import Checkpoint
logger = logging.getLogger(__name__)
......
src/ptbench/models/densenet.py
+ 2
− 1
View file @ e46a1211
......@@ -13,7 +13,8 @@ import torch.utils.data
import torchvision.models as models
import torchvision.transforms
from ..data.typing import Checkpoint, DataLoader, TransformSequence
from ..data.typing import DataLoader, TransformSequence
from .typing import Checkpoint
logger = logging.getLogger(__name__)
......
src/ptbench/models/pasa.py
+ 2
− 1
View file @ e46a1211
......@@ -13,7 +13,8 @@ import torch.optim.optimizer
import torch.utils.data
import torchvision.transforms
from ..data.typing import Checkpoint, DataLoader, TransformSequence
from ..data.typing import DataLoader, TransformSequence
from .typing import Checkpoint
logger = logging.getLogger(__name__)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment