fix import location

open-mmlab · Sep 13, 2023 · e6411f6 · e6411f6
1 parent 0d13798
commit e6411f6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/mmagic/configs/real_basicvsr/realbasicvsr_c64b20_1x30x8_8xb1_lr5e_5_150k_reds.py b/mmagic/configs/real_basicvsr/realbasicvsr_c64b20_1x30x8_8xb1_lr5e_5_150k_reds.py
@@ -9,7 +9,8 @@
 from mmengine.runner.loops import IterBasedTrainLoop
 from torch.optim.adam import Adam
 
-from mmagic.engine import DataPreprocessor, MultiOptimWrapperConstructor
+from mmagic.engine import MultiOptimWrapperConstructor
+from mmagic.models.data_preprocessors import DataPreprocessor
 from mmagic.models.editors import (RealBasicVSR, RealBasicVSRNet,
                                    UNetDiscriminatorWithSpectralNorm)
 from mmagic.models.losses import GANLoss, L1Loss, PerceptualLoss

diff --git a/mmagic/configs/real_basicvsr/realbasicvsr_wogan_c64b20_2x30x8_8xb2_lr1e_4_300k_reds.py b/mmagic/configs/real_basicvsr/realbasicvsr_wogan_c64b20_2x30x8_8xb2_lr1e_4_300k_reds.py
@@ -6,9 +6,8 @@
 
 from mmengine.config import read_base
 from mmengine.dataset.sampler import DefaultSampler, InfiniteSampler
-from mmengine.hooks import (BasicVisualizationHook, CheckpointHook,
-                            DistSamplerSeedHook, ExponentialMovingAverageHook,
-                            IterTimerHook, LoggerHook, ParamSchedulerHook)
+from mmengine.hooks import (CheckpointHook, DistSamplerSeedHook, LoggerHook,
+                            ParamSchedulerHook)
 from mmengine.optim.optimizer.optimizer_wrapper import OptimWrapper
 from mmengine.runner.loops import IterBasedTrainLoop
 from torch.optim.adam import Adam
@@ -24,6 +23,8 @@
                                         RandomVideoCompression, SetValues,
                                         UnsharpMasking)
 from mmagic.engine import MultiOptimWrapperConstructor
+from mmagic.engine.hooks import (BasicVisualizationHook,
+                                 ExponentialMovingAverageHook, IterTimerHook)
 from mmagic.evaluation import Evaluator
 from mmagic.models.data_preprocessors import DataPreprocessor
 from mmagic.models.editors import RealBasicVSR, RealBasicVSRNet