示例#1
0
 def __init__(self,
              training_service_platform: Optional[str] = None,
              **kwargs):
     super().__init__(**kwargs)
     if training_service_platform is not None:
         assert 'training_service' not in kwargs
         self.training_service = util.training_service_config_factory(
             platform=training_service_platform)
示例#2
0
文件： pytorch.py 项目： xiaowu0162/nni 
 def __init__(self,
              training_service_platform: Optional[str] = None,
              **kwargs):
     super().__init__(**kwargs)
     if training_service_platform is not None:
         assert 'training_service' not in kwargs
         self.training_service = util.training_service_config_factory(
             platform=training_service_platform)
     self.__dict__[
         'trial_command'] = 'python3 -m nni.retiarii.trial_entry py'