mirror of https://github.com/hpcaitech/ColossalAI
1cd7efc520
* [setup] refactor infer setup * [hotfix] fix infenrece behavior on 1 1 gpu * [exmaple] refactor inference examples |
||
---|---|---|
.. | ||
modeling | ||
policies | ||
__init__.py | ||
engine.py | ||
microbatch_manager.py |