File size: 417 Bytes
b84549f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.
from dataclasses import dataclass
from .common import TrainingServiceConfig
__all__ = ['AdlConfig']
@dataclass(init=False)
class AdlConfig(TrainingServiceConfig):
platform: str = 'adl'
docker_image: str = 'msranni/nni:latest'
_validation_rules = {
'platform': lambda value: (value == 'adl', 'cannot be modified')
}
|