github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-09 04:50:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

[chore] refactor profiler utils

Browse Source
This commit is contained in:
hxwang
2024-05-28 12:41:42 +00:00
parent 8547562884
commit 154720ba6e
4 changed files with 5 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/language/gpt/gemini/train_gpt_demo.py
View File

@@ -8,7 +8,8 @@ import psutil
import torch
import torch.nn as nn
from commons.model_zoo import model_builder
from commons.utils import get_data, get_profile_context, get_tflops, get_time_stamp
from commons.performance_evaluator import get_profile_context
from commons.utils import get_data, get_tflops, get_time_stamp
from packaging import version
import colossalai