github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[format] applied code formatting on changed files in pull request 4743 (#4750) 

Co-authored-by: github-actions <github-actions@github.com>
pull/4764/head
github-actions[bot] 2023-09-18 16:52:42 +08:00 committed by GitHub
parent b5f9e37c70
commit 3c6b831c26
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/language/gpt/experiments/pipeline_parallel/train_gpt_pp.py
View File

@ -3,6 +3,7 @@ import time
from functools import partial from functools import partial
import torch import torch
from model_zoo import model_builder
from torch import nn from torch import nn
from tqdm import tqdm from tqdm import tqdm
@ -18,7 +19,6 @@ from colossalai.legacy.pipeline.rpc._pipeline_schedule import FillDrainPipelineE
from colossalai.legacy.pipeline.rpc.utils import rpc_run from colossalai.legacy.pipeline.rpc.utils import rpc_run
from colossalai.logging import disable_existing_loggers, get_dist_logger from colossalai.logging import disable_existing_loggers, get_dist_logger
from colossalai.nn.optimizer import HybridAdam from colossalai.nn.optimizer import HybridAdam
from model_zoo import model_builder
def parse_args(): def parse_args():