Fix more typing

This commit is contained in:
henryruhs
2025-01-14 12:48:43 +01:00
parent 2bbba3563b
commit 23ac63d55b
3 changed files with 7 additions and 3 deletions
+3 -1
View File
@@ -1,7 +1,9 @@
from typing import Tuple
import torch
import torch.nn as nn
from .typing import IDEmbedding, TargetAttributes, Tensor, Tuple
from .typing import IDEmbedding, TargetAttributes, Tensor
class AdaptiveEmbeddingIntegrationNetwork(nn.Module):
+2 -1
View File
@@ -1,8 +1,9 @@
import configparser
from typing import Tuple
import torch
from .typing import Tensor, Tuple
from .typing import Tensor
CONFIG = configparser.ConfigParser()
CONFIG.read('config.ini')
+2 -1
View File
@@ -1,5 +1,6 @@
import configparser
import os
from typing import Tuple
import cv2
import numpy
@@ -17,7 +18,7 @@ from .data_loader import DataLoaderVGG, read_image
from .discriminator import MultiscaleDiscriminator
from .generator import AdaptiveEmbeddingIntegrationNetwork
from .helper import L2_loss, hinge_loss
from .typing import Batch, DiscriminatorOutputs, IDEmbedding, LossDict, TargetAttributes, Tensor, Tuple
from .typing import Batch, DiscriminatorOutputs, IDEmbedding, LossDict, TargetAttributes, Tensor
CONFIG = configparser.ConfigParser()
CONFIG.read('config.ini')