abhishekkrthakur / tez

Tez is a super-simple and lightweight Trainer for PyTorch. It also comes with many utils that you can use to tackle over 90% of deep learning projects in PyTorch.
Apache License 2.0
1.16k stars 145 forks source link

Integrated sandesh with tez! #3

Closed devanshu125 closed 2 years ago

devanshu125 commented 3 years ago

The whole idea behind this integration was that I was pretty bored by just sitting and staring at the screen while training models with tez. So, I thought that it would be great to integrate @abhishekkrthakur 's project sandesh, with tez. I have also added functionality to get notifications for EarlyStopping by just setting es_notifications to True. Just import SendSandesh from tez.callbacks , specify the relevant arguments and you are good to go! (P.S. Don't forget the webhook) sandesh_image

gauravbrills commented 2 years ago

Wouldn't it be better to do similar by integrating with W&B for example