zhaocq-nlp
commented
2 years ago The versions are as follows： python=3.6 tensorflow-gpu==2.4.0 neurst==0.1.0
Run command: python3 -m neurst.cli.run_exp --config_paths data/training_args.yml,data/translation_bpe.yml --hparams_set waitk_transformer_base --model_dir checkpoints/waitk_enzh_base --task WaitkTranslation --wait_k 5
After running, the following information appears： I0127 13:06:57.520065 140639346566976 configurable.py:296] Saving model configurations to directory: checkpoints/waitk_enzh_base 2022-01-27 13:06:57.549626: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2) 2022-01-27 13:06:57.550100: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2300000000 Hz
It has been stuck here, and I am asking if the environment configuration is inappropriate? Please recommend a suitable environment version
Could please try tf 2.4.1? We tested the code and succeeded on our machine.
akk-123
The versions are as follows： python=3.6 tensorflow-gpu==2.4.0 neurst==0.1.0
Run command: python3 -m neurst.cli.run_exp \ --config_paths data/training_args.yml,data/translation_bpe.yml \ --hparams_set waitk_transformer_base \ --model_dir checkpoints/waitk_enzh_base \ --task WaitkTranslation \ --wait_k 5
After running, the following information appears： I0127 13:06:57.520065 140639346566976 configurable.py:296] Saving model configurations to directory: checkpoints/waitk_enzh_base 2022-01-27 13:06:57.549626: I tensorflow/compiler/mlir/mlir_graph_optimization_pass.cc:116] None of the MLIR optimization passes are enabled (registered 2) 2022-01-27 13:06:57.550100: I tensorflow/core/platform/profile_utils/cpu_utils.cc:112] CPU Frequency: 2300000000 Hz
It has been stuck here, and I am asking if the environment configuration is inappropriate? Please recommend a suitable environment version