github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-04-28 11:45:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

[NFC] polish colossalai/fx/profiler/experimental/profiler_module/embedding.py code style (#3256)

Browse Source 
Co-authored-by: Minghao Huang <huangminghao@luchentech.com>
This commit is contained in:
dayellow 2023-03-27 18:43:43 +08:00 committed by binmakeswell
parent 0fbadce79c
commit 204ca2f09a
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/fx/profiler/experimental/profiler_module/embedding.py
View File

@ -1,5 +1,7 @@
from typing import Tuple
import torch
from ..registry import meta_profiler_module