def test_vgg_11_bn( download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, ): model_constructor( vgg_11bn, download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, )
def test_inception_v3( download, framework, repo, dataset, training_scheme, optim_name, optim_category, optim_target, ): model_constructor( inception_v3, download, framework, repo, dataset, training_scheme, optim_name, optim_category, optim_target, )
def test_efficientnet_b4( download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, ): model_constructor( efficientnet_b4, download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, )
def test_vgg_11( download, framework, repo, dataset, training_scheme, optim_name, optim_category, optim_target, ): model_constructor( vgg_11, download, framework, repo, dataset, training_scheme, optim_name, optim_category, optim_target, )
def test_ssd_resnet50_300_extended( download, framework, repo, dataset, training_scheme, optim_name, optim_category, optim_target, ): model_constructor( ssd_resnet50_300, download, framework, repo, dataset, training_scheme, optim_name, optim_category, optim_target, )
def test_mobilenet_v2( download, framework, repo, dataset, training_scheme, optim_name, optim_category, optim_target, ): model_constructor( mobilenet_v2, download, framework, repo, dataset, training_scheme, optim_name, optim_category, optim_target, )
def test_yolo_v3( download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, ): model_constructor( yolo_v3, download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, )
def test_efficientnet_b0( download, framework, repo, dataset, training_scheme, optim_name, optim_category, optim_target, ): model_constructor( efficientnet_b0, download, framework, repo, dataset, training_scheme, optim_name, optim_category, optim_target, )
def test_resnet_34( download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, ): model_constructor( resnet_34, download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, )
def test_mobilenet_v1( download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, ): model_constructor( mobilenet_v1, download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, )
def test_ssd_resnet50_300( download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, ): model_constructor( ssd_resnet50_300, download, framework, repo, dataset, training_scheme, sparse_name, sparse_category, sparse_target, )