wait-k running?

[Vincent131499]

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

[zhaocq-nlp]

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]

@zhaocq-nlp when will release pytorch version?