Coverage for /home/ubuntu/flatiron/python/flatiron/torch/loss.py: 100%

1from flatiron.core.types import Getter # noqa F401 

2 

3from torch.nn import Module # noqa: F401 

4 

5import flatiron.torch.tools as fi_torchtools 

6# ------------------------------------------------------------------------------ 

7 

8 

9def get(config): 

10 # type: (Getter) -> Module 

11 ''' 

12 Get function from this module. 

13 

14 Args: 

15 config (dict): Loss config. 

16 

17 Returns: 

18 function: Module function. 

19 ''' 

20 return fi_torchtools.get(config, __name__, 'torch.nn.modules.loss')