def test_mmoe_gpu(self):
"""test multitask.mmoe with gpu."""
self.yaml_config_name = "models/multitask/mmoe/config.yaml"
sed_gpu_cmd = "sed -i 's/device:.*cpu/device: gpu/g' {}".format(self.yaml_config_name)
utils.cmd_shell(sed_gpu_cmd)
self.run_yaml(generate=False, cuda_devices="0")
built_in.equals(self.pro.returncode, 0, self.err_msg)
built_in.not_contains(self.err, 'Traceback', self.err_msg)
def test_youtube_dnn_gpu(self):
"""test recall.youtube_dnn with gpu."""
self.yaml_config_name = "models/recall/youtube_dnn/config.yaml"
sed_gpu_cmd = "sed -i 's/device:.*cpu/device: gpu/g' {}".format(self.yaml_config_name)
utils.cmd_shell(sed_gpu_cmd)
self.run_yaml(generate=False, cuda_devices="0")
built_in.equals(self.pro.returncode, 0, self.err_msg)
built_in.not_contains(self.err, 'Traceback', self.err_msg)
def test_AutoInt(self):
"""test rank.fibinet"""
self.yaml_config_name = "models/rank/AutoInt/config.yaml"
verify_epochs = "sed -i 's/epochs:.*2/epochs: 1/g' {}".format(self.yaml_config_name)
utils.cmd_shell(verify_epochs)
self.run_yaml(generate=False)
built_in.equals(self.pro.returncode, 0, self.err_msg)
built_in.not_contains(self.err, 'Traceback', self.err_msg)
def test_tagspace_gpu(self):
"""test contentunderstanding.tagspace with gpu."""
self.yaml_config_name = "models/contentunderstanding/tagspace/config.yaml"
sed_gpu_cmd = "sed -i 's/device:.*cpu/device: gpu/g' {}".format(self.yaml_config_name)
utils.cmd_shell(sed_gpu_cmd)
self.run_yaml(generate=False, cuda_devices="0")
built_in.equals(self.pro.returncode, 0, self.err_msg)
built_in.not_contains(self.err, 'Traceback', self.err_msg)
def test_mode_list_ps_local_cluster_c2(self):
"""test mode list has one element and the runner is local cluster.
"""
utils.cmd_shell("rm -rf logs")
self.yaml_config_name = sys._getframe().f_code.co_name + '.yaml'
self.yaml_content["runner"][0]["class"] = "local_cluster_train"
self.run_yaml()
built_in.equals(self.pro.returncode, 0, self.err_msg)
built_in.not_contains(self.err, 'Traceback', self.err_msg)
def test_fibinet_gpu(self):
"""test rank.fibinet"""
self.yaml_config_name = "models/rank/fibinet/config.yaml"
verify_epochs = "sed -i 's/epochs:.*4/epochs: 1/g' {}".format(
self.yaml_config_name)
sed_gpu_cmd = "sed -i 's/device:.*cpu/device: gpu/g' {}".format(
self.yaml_config_name)
utils.cmd_shell(sed_gpu_cmd)
utils.cmd_shell(verify_epochs)
self.run_yaml(generate=False, cuda_devices="0")
built_in.equals(self.pro.returncode, 0, self.err_msg)
built_in.not_contains(self.err, 'Traceback', self.err_msg)
def setUp(self):
"""do something for each cases."""
utils.cmd_shell("rm -rf logs")
utils.cmd_shell("rm -rf increment*")
utils.cmd_shell("rm -rf inference*")
utils.cmd_shell("cp -r PaddleRec/models ./")
def setUp(self):
"""do something for each cases."""
utils.cmd_shell("rm -rf logs")
