mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-12 07:02:22 +00:00
12 lines
190 B
Python
12 lines
190 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
|