예제 #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'