コード例 #1
0
ファイル: pytorch.py プロジェクト: hmrg-grmh/microsoft-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)
コード例 #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'