diff --git a/mainInformer.py b/mainInformer.py index 137edb5..744989a 100644 --- a/mainInformer.py +++ b/mainInformer.py @@ -3,7 +3,7 @@ import os import mindspore import mindspore.context as context from mindspore.train.serialization import load_checkpoint, save_checkpoint -1 + from exp.exp_informer import Exp_Informer context.set_auto_parallel_context(enable_auto_memory=True)