(self, **kwargs)
| 36 | structure = State(autoload=False) |
| 37 | |
| 38 | def __init__(self, **kwargs): |
| 39 | self.load('structure', **kwargs) |
| 40 | |
| 41 | cmd = kwargs.get('cmd', {}) |
| 42 | if 'name' not in cmd: |
| 43 | cmd['name'] = self.structure.model_name |
| 44 | |
| 45 | self.load_all(**kwargs) |
| 46 | self.distributed = cmd.get('distributed', False) |
| 47 | self.local_rank = cmd.get('local_rank', 0) |
| 48 | |
| 49 | if cmd.get('validate', False): |
| 50 | self.load('validation', **kwargs) |
| 51 | else: |
| 52 | self.validation = None |
| 53 |