mirror of
https://github.com/PaddlePaddle/PaddleClas.git
synced 2025-06-03 21:55:06 +08:00
15 lines
337 B
Python
15 lines
337 B
Python
|
from ..engine import build_engine
|
||
|
|
from ..utils import config
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
args = config.parse_args()
|
||
|
|
config_dict = config.get_config(
|
||
|
|
args.config, overrides=args.override, show=False)
|
||
|
|
config_dict.profiler_options = args.profiler_options
|
||
|
|
engine = build_engine(config_dict)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|