def test_lstm(self): args = [ "--framework=pytorch", "--problem=lstm", "--problem_size=4,4,4", "--batch_size=4", "--nb_epoch=1", "--mode=inference", ] run(args)
def test_bad_mode(self): with self.assertRaises(AssertionError): args = [ "--framework=pytorch", "--problem=conv1d", "--problem_size=4,4,4", "--batch_size=4", "--mode=depeche", ] run(args)
def test_attention(self): args = [ "--framework=pytorch", "--problem=attention", "--problem_size=2,2,4", "--cnt_heads=2", "--batch_size=1", "--nb_epoch=1", "--mode=inference", ] run(args)
def test_vgg19(self): run(["--problem=vgg19", *self.imgnet_args])
def test_xception(self): run(["--problem=xception", *self.imgnet_args])
def test_ncf_inference(self): run(self.args + ["--mode=inference"])
def test_ssd300(self): run(self.args)
def test_resnet50(self): run(["--problem=resnet50"] + self.imgnet_args)
def test_cosmoflow(self): run(self.args)
def test_vgg16(self): run(self.args + ["--mode=training"])
def test_vgg16_inference(self): run(["--problem=vgg16", "--mode=inference", *self.imgnet_args])
def test_cnn2d_training(self): run(self.args + ["--mode=training"])
def test_cnn2d_inference(self): run(self.args + ["--mode=inference"])
def test_bert_training(self): run(self.args + ["--mode=training"])
def test_no_framework(self): with self.assertRaises(Exception): run([])
def test_res10ssd(self): run(self.args)
def test_resnet50_inference(self): run(self.args + ["--mode=inference"])
def test_vgg16_inference(self): run(self.args + ["--mode=inference"])
def test_cosmoflow_inference(self): run(self.args + ["--mode=inference"])
def test_bert(self): run(self.args)
def test_resnet50_inference(self): run(["--problem=resnet50", "--mode=inference"] + self.imgnet_args)
def test_ncf(self): run(self.args + ["--mode=training"])
def test_conv3d(self): run(self.args + ["--problem=conv3d", "--problem_size=2,4,4,4,4"])
def test_no_problem(self): with self.assertRaises(Exception): run(["--framework=pytorch"])