<p dir="auto">I use examples/pytorch/translation/run_translation.py fine-tune mbart-large-cc25 on my datasets, it automatically runs on CPU.  I have 2 GPU, but only one is Nvidia.It is RTX 2080super.</p>
<p dir="auto">python main.py <br>
--model_name_or_path facebook/mbart-large-cc25 <br>
--do_train  <br>
--do_eval <br>
--source_lang en_XX <br>
--target_lang zh_CN <br>
--train_file /data/2WangHongyu/bioNMT_WHY/train.json <br>
--validation_file /data/2WangHongyu/bioNMT_WHY/dev.json <br>
--output_dir /output <br>
--per_device_train_batch_size=4 <br>
--per_device_eval_batch_size=4 <br>
--overwrite_output_dir <br>
--predict_with_generate <br>
--cache_dir /model/2WangHongyu/mbart-large</p>
<p dir="auto">-<code class="notranslate">transformers</code> version: 4.5.0</p>
<ul dir="auto">
<li>Platform: linux</li>
<li>Python version: 3.8</li>
<li>PyTorch version (GPU?): 1.7.1</li>
<li>Tensorflow version (GPU?):</li>
<li>Using GPU in script?: yes</li>
<li>Using distributed or parallel set-up in script?:yes</li>
</ul>
<p dir="auto"><a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/patrickvonplaten/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/patrickvonplaten">@patrickvonplaten</a>, <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="/users/patil-suraj/hovercard" data-octo-click="hovercard-link-click" data-octo-dimensions="link_type:self" href="https://github.com/patil-suraj">@patil-suraj</a><br>
Models: mbart-large-50</p>