mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-05-30 10:55:24 +00:00
0fedef4f3c
* integrated parallel layers for ease of building models * integrated 2.5d layers * cleaned codes and unit tests * added log metric by step hook; updated imagenet benchmark; fixed some bugs * reworked initialization; cleaned codes Co-authored-by: BoxiangW <45734921+BoxiangW@users.noreply.github.com>
7 lines
137 B
Python
7 lines
137 B
Python
from .layer import *
|
|
from .loss import *
|
|
from .lr_scheduler import *
|
|
from .metric import *
|
|
from .model import *
|
|
from .optimizer import *
|