update

muxuezzz · Jul 29, 2021 · 325d5cc · 325d5cc
1 parent ed38719
commit 325d5cc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/advchain/augmentor/__init__.py b/advchain/augmentor/__init__.py
@@ -1,6 +1,7 @@
-from adv_affine import AdvAffine
-from adv_bias import AdvBias
-from adv_morph import AdvMorph
-from adv_noise import AdvNoise
-from adv_transformation_base import AdvTransformationBase
-from adv_compose_solver import AdvComposeSolver
+from adv_transformation_base import *
+from adv_transformation_base import *
+from adv_affine import *
+from adv_bias import *
+from adv_morph import *
+from adv_noise import *
+from adv_compose_solver import *
diff --git a/advchain/augmentor/adv_compose_solver.py b/advchain/augmentor/adv_compose_solver.py
@@ -2,9 +2,8 @@
 import numpy as np
 import torch
 import torch.nn.functional as F
-
-from advchain.common.loss import calc_segmentation_consistency  # noqa
-from advchain.common.utils import _disable_tracking_bn_stats, set_grad  # noqa
+from advchain.common.loss import calc_segmentation_consistency ##noqa
+from advchain.common.utils import _disable_tracking_bn_stats, set_grad
 
 
 class ComposeAdversarialTransformSolver(object):