shreshthtuli/aiotbench repository overview

To run use the following command:

docker run shreshthtuli/aiotbench 'python3' 'main.py' '<model>'

where <model> can be one of resnet18, resnet34, squeezenet1_0, mobilenet_v2, mnasnet1_0, googlenet or resnext50_32x4d.