Ok thanks for the reply, I'll give it a try myself ! Huge tx again for you vast.ai template it works perfectly. If you plan to evolve it my 2 feature requests would be :
- Suggest the best checkpoints based on loss value
- Ability to input some prompts and generate samples to be displayed in the UI and uploaded to the cloud at the end of the training.