From 1ea7cac723e49f94acc3396f00e4ed48290cd3e8 Mon Sep 17 00:00:00 2001 From: darth_willy Date: Wed, 16 Jan 2019 15:39:50 -0400 Subject: [PATCH] removed shard_to_cpu --- g2p_seq2seq/g2p_trainer_utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/g2p_seq2seq/g2p_trainer_utils.py b/g2p_seq2seq/g2p_trainer_utils.py index a106de9e..b3153d19 100644 --- a/g2p_seq2seq/g2p_trainer_utils.py +++ b/g2p_seq2seq/g2p_trainer_utils.py @@ -214,7 +214,6 @@ def create_run_config(hp, params): keep_checkpoint_every_n_hours=params.keep_checkpoint_every_n_hours, num_gpus=params.worker_gpu, gpu_order=params.gpu_order, - shard_to_cpu=params.locally_shard_to_cpu, num_async_replicas=params.worker_replicas, enable_graph_rewriter=params.experimental_optimize_placement, gpu_mem_fraction=params.worker_gpu_memory_fraction,