Commit 9be415c
committed
Adjust import statements
1 parent 5d08bb4 commit 9be415c
1 file changed
Lines changed: 4 additions & 6 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -32,11 +32,10 @@ | |||
| 32 | 32 | ||
| 33 | 33 | from plasma.conf import conf | |
| 34 | 34 | from plasma.models.loader import Loader | |
| 35 | - from plasma.preprocessor.normalize import Normalizer, Augmentator | ||
| 35 | + from plasma.preprocessor.normalize import Normalizer | ||
| 36 | + from plasma.preprocessor.augment import Augmentator | ||
| 36 | 37 | from plasma.preprocessor.preprocess import guarantee_preprocessed | |
| 37 | 38 | ||
| 38 | - import pdb | ||
| 39 | - | ||
| 40 | 39 | if conf['model']['shallow']: | |
| 41 | 40 | print("Shallow learning using MPI is not supported yet. set conf['model']['shallow'] to false.") | |
| 42 | 41 | exit(1) | |
@@ -83,12 +82,11 @@ | |||
| 83 | 82 | ||
| 84 | 83 | ||
| 85 | 84 | print("normalization",end='') | |
| 86 | - pdb.set_trace() | ||
| 87 | 85 | raw_normalizer = Normalizer(conf) | |
| 88 | 86 | raw_normalizer.train() | |
| 89 | 87 | is_inference= False | |
| 90 | - normalizer = Augmentator(Normalizer,is_inference,conf) | ||
| 91 | - loader = Loader(conf,nn) | ||
| 88 | + normalizer = Augmentator(raw_normalizer,is_inference,conf) | ||
| 89 | + loader = Loader(conf,normalizer) | ||
| 92 | 90 | print("...done") | |
| 93 | 91 | ||
| 94 | 92 | if not only_predict: | |
0 commit comments