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

#869 Add DagsHub Logger to Super Gradients

Merged
Ghost merged 1 commits into Deci-AI:master from timho102003:dagshub_logger
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
  1. from typing import Union, Mapping
  2. from super_gradients.common.factories.base_factory import BaseFactory
  3. from super_gradients.common.factories.list_factory import ListFactory
  4. from super_gradients.common.registry.registry import PROCESSINGS
  5. class ProcessingFactory(BaseFactory):
  6. def __init__(self):
  7. super().__init__(PROCESSINGS)
  8. def get(self, conf: Union[str, dict]):
  9. if isinstance(conf, Mapping) and "ComposeProcessing" in conf:
  10. conf["ComposeProcessing"]["processings"] = ListFactory(ProcessingFactory()).get(conf["ComposeProcessing"]["processings"])
  11. return super().get(conf)
Discard
Tip!

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