From e30d782bf4bd765d5ddd295635b7f85c6817d800 Mon Sep 17 00:00:00 2001 From: Yina Chen <33650826+cyita@users.noreply.github.com> Date: Mon, 15 Jun 2020 10:55:31 +0800 Subject: [PATCH] Temporarily remove mxnet trainer list input test (#2446) --- .../learn/mxnet/test_mxnet_ray_xshards.py | 38 +++++++++---------- 1 file changed, 19 insertions(+), 19 deletions(-) diff --git a/pyzoo/test/zoo/orca/learn/mxnet/test_mxnet_ray_xshards.py b/pyzoo/test/zoo/orca/learn/mxnet/test_mxnet_ray_xshards.py index 6c7af85cd23..03de6dbbbd7 100644 --- a/pyzoo/test/zoo/orca/learn/mxnet/test_mxnet_ray_xshards.py +++ b/pyzoo/test/zoo/orca/learn/mxnet/test_mxnet_ray_xshards.py @@ -125,25 +125,25 @@ def test_xshards_gluon(self): num_workers=2) trainer.train(nb_epoch=2) - def test_xshard_list(self): - # prepare data - resource_path = os.path.join(os.path.split(__file__)[0], "../../../resources") - self.ray_ctx = get_ray_ctx() - train_file_path = os.path.join(resource_path, - "orca/learn/multi_input_json/train_data_list.json") - train_data_shard = zoo.orca.data.pandas.read_json(train_file_path, self.ray_ctx, - orient="records", lines=False) - train_data_shard.transform_shard(prepare_data_list) - test_file_path = os.path.join(resource_path, - "orca/learn/multi_input_json/test_data_list.json") - test_data_shard = zoo.orca.data.pandas.read_json(test_file_path, self.ray_ctx, - orient="records", lines=False) - test_data_shard.transform_shard(prepare_data_list) - config = create_trainer_config(batch_size=32, log_interval=1, seed=42) - trainer = MXNetTrainer(config, train_data_shard, get_gluon_model, get_loss, - validation_metrics_creator=get_gluon_metrics, - test_data=test_data_shard, eval_metrics_creator=get_gluon_metrics) - trainer.train(nb_epoch=2) + # def test_xshard_list(self): + # # prepare data + # resource_path = os.path.join(os.path.split(__file__)[0], "../../../resources") + # self.ray_ctx = get_ray_ctx() + # train_file_path = os.path.join(resource_path, + # "orca/learn/multi_input_json/train_data_list.json") + # train_data_shard = zoo.orca.data.pandas.read_json(train_file_path, self.ray_ctx, + # orient="records", lines=False) + # train_data_shard.transform_shard(prepare_data_list) + # test_file_path = os.path.join(resource_path, + # "orca/learn/multi_input_json/test_data_list.json") + # test_data_shard = zoo.orca.data.pandas.read_json(test_file_path, self.ray_ctx, + # orient="records", lines=False) + # test_data_shard.transform_shard(prepare_data_list) + # config = create_trainer_config(batch_size=32, log_interval=1, seed=42) + # trainer = MXNetTrainer(config, train_data_shard, get_gluon_model, get_loss, + # validation_metrics_creator=get_gluon_metrics, + # test_data=test_data_shard, eval_metrics_creator=get_gluon_metrics) + # trainer.train(nb_epoch=2) if __name__ == "__main__":