Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel
Resources  
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Explore Pricing
Product  
Data Engine LLMs Platform Enterprise
Connect to Dagshub Discord channel
Register
Login

Deci-AI
/
super-gradients
connected to https://github.com/Deci-AI/super-gradients.git

Watch 4
Issues 62 Pull Requests 44 Discussions 0

#647 Feature/sg 573 Integrate new EMA decay schedules

Merged
Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/SG-573-Integrate-EMA
Conversation 0 Files changed 11 Commits 1
File Compare
super-gradients
/
tests
/
unit_tests
/
kd_ema_test.py
View left file View right file 
@@ -34,6 +34,7 @@ class KDEMATest(unittest.TestCase):
 
             "greater_metric_to_watch_is_better": True,
 
             "greater_metric_to_watch_is_better": True,
 
             "average_best_models": False,
 
             "average_best_models": False,
 
             "ema": True,
 
             "ema": True,
 
+            "ema_params": {"decay_type": "constant", "decay": 0.999},
 
         }
 
         }
 
 
 
     def test_teacher_ema_not_duplicated(self):
 
     def test_teacher_ema_not_duplicated(self):
Discard
Prev
Next
Tip!

Press p or to see the previous file or, n or to see the next file