diff --git a/fluid/neural_machine_translation/transformer/train.py b/fluid/neural_machine_translation/transformer/train.py index d82a233a6e1af1b198ebe2ff8cec89d500f169b7..4b10c26e3e5b3b99adac19784b519829ce618805 100644 --- a/fluid/neural_machine_translation/transformer/train.py +++ b/fluid/neural_machine_translation/transformer/train.py @@ -225,6 +225,7 @@ def main(): fluid.CUDAPlace(place_id)) for var_name in util_input_dict: + print var_name, local_scope.find_var(var_name) local_scope.find_var(var_name).get_tensor().set(util_input_dict[var_name], fluid.CUDAPlace(place_id))