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