コード例 #1
0
ファイル: swarm_agent.py プロジェクト: jmbjr/Saltie
 def __init__(self, name, team, index):
     super().__init__(name, team, index)
     sys.path.insert(
         0, get_repo_directory())  # this is for separate process imports
     self.logger = get_logger(name)
     self.manager_path = None
     self.model_path = None
     self.load_model = None
コード例 #2
0
    def __init__(self, agent_metadata_queue, quit_event):
        super().__init__(agent_metadata_queue, quit_event)
        sys.path.insert(
            0, get_repo_directory())  # this is for separate process imports
        self.logger = get_logger('base_hive_mgr')

        self.manager = self.setup_manager()
        self.game_memory = self.manager.Memory()
        self.actor_model = self.get_model()
        self.shared_model_handle = self.get_shared_model_handle()
        self.model_path = None
        self.load_model = None

        self.setup_trainer()
コード例 #3
0
ファイル: teacher_agent.py プロジェクト: jmbjr/Saltie
 def load_config(self, config_object_header: ConfigHeader):
     super().load_config(config_object_header)
     teacher_path = config_object_header.get('teacher_path')
     self.teacher = ExternalClassWrapper(
         os.path.join(get_repo_directory(), teacher_path),
         BaseAgent).get_loaded_class()(self.name, self.team, self.index)
コード例 #4
0
 def get_file_path(self):
     return get_repo_directory() + '/trainer/weights/' + str(
         type(self.model).__name__) + '.mdl'
コード例 #5
0
 def get_file_path(self):
     return os.path.join(get_repo_directory(), self.model_path)
コード例 #6
0
ファイル: base_model_holder.py プロジェクト: skoocda/Saltie
 def get_file_path(self):
     return get_repo_directory(
     ) + '/trainer/weights/' + self.get_model_name() + '.mdl'