Reinforcement learning for machine translation use FairSeq toolkit. This image support Weight And Biases ,Fairseq image docker https://hub.docker.com/layers/nvip1204/fairseq/c600667/images/sha256-9dc185c204ccda8fde3f17b9ab72a7a05320c9c7f68bf41ec66ba78fc5e6fbe6?context=repo
Example run: https://wandb.ai/nvip1204/test_docker/runs/pbwjnb68?workspace=user-nvip1204 Command to run docker image: docker run --ipc=host --gpus all -it --rm -v /home/nvip/data/JAIST/luan_van/RL_transformer/:/workspace/test fairseq_c600667