Cleanup - Githubissues

kaylode / theseus

General template for most Pytorch projects

MIT License

34 stars 6 forks source link

Cleanup #36

Closed kaylode closed 2 years ago

kaylode commented 2 years ago

Add BasePipeline, BaseTestPipeline for efficient reusability
Convert classification metrics to enable multilabel tasks
Add HuggingFace model for text classification
Fix bugs in logging, loading, trainer, configs
Allow scheduler to be None, add save/load scheduler
Add shutdown flags to trainer, so that callbacks can stop training anytime