From 2d40759a5351aa08866d37a39b5f141e7a11e0eb Mon Sep 17 00:00:00 2001 From: digger yu Date: Wed, 28 Jun 2023 15:29:44 +0800 Subject: [PATCH] fix #3852 path error (#4058) --- examples/language/gpt/titans/train_gpt.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/language/gpt/titans/train_gpt.py b/examples/language/gpt/titans/train_gpt.py index 66225d6c8..6be0b9e8d 100644 --- a/examples/language/gpt/titans/train_gpt.py +++ b/examples/language/gpt/titans/train_gpt.py @@ -15,7 +15,7 @@ from colossalai.nn import LinearWarmupLR from colossalai.trainer import Trainer, hooks from colossalai.utils import colo_set_process_memory_fraction, is_using_pp from colossalai.utils.timer import MultiTimer -from colossalai.zero.init_ctx import ZeroInitContext +from colossalai.zero.legacy.init_ctx import ZeroInitContext def calc_local_model_size(model: torch.nn.Module):