bloomsburyai/question-generation

Unable to save the model

Closed this issue · 2 comments

ayush2051 commented

@tomhosking I am unable to save the checkpoints while training the model. I guess m unable to figure out what should be in the restore path. Can u help me in this regard?

tomhosking commented

Checkpoints are saved to the location specified by the --model_dir flag, default is ./models. You may need to create this directory?

  • 👍1
ayush2051 commented

@tomhosking thanks a lot for the reply. Got it fixed.

Checkpoints are saved to the location specified by the --model_dir flag, default is ./models. You may need to create this directory?