PreSumm
PreSumm copied to clipboard
visible gpu setting and bert model download
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:
- how do I know the download of bert is completed and I can stop the training session not using gpu?
- what happens if I just train with gpu = 1 from the very beginning?
- 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.
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
I have same questions with you, did you solve it?