PreSumm icon indicating copy to clipboard operation
PreSumm copied to clipboard

visible gpu setting and bert model download

Open wailoktam opened this issue 5 years ago • 2 comments

hi, I am sort of puzzled by the instruction on setting visible gpus to -1, kill the training process and rerun. Here are my questions:

  1. how do I know the download of bert is completed and I can stop the training session not using gpu?
  2. what happens if I just train with gpu = 1 from the very beginning?
  3. can I just download the bert model, put it somewhere for the program to find it and avoid the switch?

I am asking these questions because training with gpus = -1 and testing with gpus = 1 would result in error, not to mention the problem that training with gpus = -1 would take ages. Sticking with gpus = 1 seems more straightforward and I wonder how to do it.

wailoktam avatar Jun 16 '20 02:06 wailoktam

This is only the case if you have more than 1 gpu. If you only use 1 gpu, try with gpus = 0. Then it will download the model and start training without any issues

SebastianVeile avatar Jun 18 '20 09:06 SebastianVeile

I have same questions with you, did you solve it?

Dejavu2019 avatar Oct 08 '20 07:10 Dejavu2019