Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "115bcc0b416cd050326cbd6e42b1cc00a8053a82"
Unverified Commit 2d40759a authored by digger yu's avatar digger yu Committed by GitHub
Browse files

fix #3852 path error (#4058)

parent 1ee947f6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/language/gpt/titans/train_gpt.py
View file @ 2d40759a
......@@ -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):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment