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"])
