search

jadore801120 / attention-is-all-you-need-pytorch

A PyTorch implementation of the Transformer model in "Attention is All You Need".
MIT License
8.82k stars 1.98k forks source link

What command can continue running program? #35

Closed cong1 closed 4 years ago

cong1 commented 6 years ago

Hi,thanks for your sharing. My program has broken down。 What command can continue running program at GPU?

CUDA_VISIBLE_DEVICES=3 python train.py -data data/multi30k.atok.low.pt -save_model trained -save_mode best -proj_share_weight

jadore801120 commented 4 years ago

You can follow the translate.py to load a trained model weight. Hope this helps