mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-12-02 11:06:39 +00:00
14 lines
228 B
Python
14 lines
228 B
Python
from . import (
|
|
bert,
|
|
gpt2,
|
|
hanging_param_model,
|
|
inline_op_model,
|
|
nested_model,
|
|
repeated_computed_layer,
|
|
resnet,
|
|
simple_net,
|
|
)
|
|
from .utils import run_fwd_bwd
|
|
|
|
from . import albert # isort:skip
|