#399 Feature/sg 326 replace function with class in architectures

Merged

Ghost merged 1 commits into Deci-AI:master from deci-ai:feature/sg-326-replace_function_with_class_in_architectures

  
1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

            from typing import Union, Mapping
from omegaconf import ListConfig
from super_gradients.common.factories.base_factory import BaseFactory
from super_gradients.common.factories.list_factory import ListFactory
from super_gradients.training.transforms import TRANSFORMS
class TransformsFactory(BaseFactory):
    def __init__(self):
        super().__init__(TRANSFORMS)
    def get(self, conf: Union[str, dict]):
        # SPECIAL HANDLING FOR COMPOSE
        if isinstance(conf, Mapping) and 'Compose' in conf:
            conf['Compose']['transforms'] = ListFactory(TransformsFactory()).get(conf['Compose']['transforms'])
        elif isinstance(conf, (list, ListConfig)):
            conf = ListFactory(TransformsFactory()).get(conf)
        return super().get(conf)

Tip!

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

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

#399 Feature/sg 326 replace function with class in architectures

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