17 lines
250 B
Python
17 lines
250 B
Python
import click
|
|
|
|
from benchmark.cifar10.train import train
|
|
from benchmark.cifar10.infer import infer
|
|
|
|
|
|
@click.group()
|
|
def cli():
|
|
pass
|
|
|
|
|
|
cli.add_command(train, name='train')
|
|
cli.add_command(infer, name='infer')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
cli()
|