コード例 #1
0
ファイル: ffnn.py プロジェクト: ppoulin91/learn2track
    def save(self, path):
        savedir = smartutils.create_folder(pjoin(path, type(self).__name__))
        smartutils.save_dict_to_json_file(pjoin(savedir, "hyperparams.json"), self.hyperparameters)

        params = {param.name: param.get_value() for param in self.parameters}
        assert len(self.parameters) == len(params)  # Implies names are all unique.
        np.savez(pjoin(savedir, "params.npz"), **params)
コード例 #2
0
    def save(self, path):
        savedir = utils.create_folder(pjoin(path, "model"))
        hyperparameters = {'input_size': self.input_size,
                           'output_size': self.output_size}
        utils.save_dict_to_json_file(pjoin(savedir, "meta.json"), {"name": self.__class__.__name__})
        utils.save_dict_to_json_file(pjoin(savedir, "hyperparams.json"), hyperparameters)

        params = {param.name: param.get_value() for param in self.parameters}
        np.savez(pjoin(savedir, "params.npz"), **params)
コード例 #3
0
ファイル: gru_msp.py プロジェクト: ppoulin91/learn2track
    def save(self, path):
        super().save(path)

        savedir = smartutils.create_folder(pjoin(path, type(self).__name__))
        state = {
            "version": 1,
            "_srng_rstate": self.srng.rstate,
            "_srng_state_updates": [state_update[0].get_value() for state_update in self.srng.state_updates],
        }

        np.savez(pjoin(savedir, "state.npz"), **state)
コード例 #4
0
    def save(self, path):
        savedir = utils.create_folder(pjoin(path, "model"))
        hyperparameters = {
            'input_size': self.input_size,
            'output_size': self.output_size
        }
        utils.save_dict_to_json_file(pjoin(savedir, "meta.json"),
                                     {"name": self.__class__.__name__})
        utils.save_dict_to_json_file(pjoin(savedir, "hyperparams.json"),
                                     hyperparameters)

        params = {param.name: param.get_value() for param in self.parameters}
        np.savez(pjoin(savedir, "params.npz"), **params)
コード例 #5
0
    def save(self, path):
        super().save(path)

        savedir = smartutils.create_folder(pjoin(path, type(self).__name__))
        state = {
            "version":
            1,
            "_srng_rstate":
            self.srng.rstate,
            "_srng_state_updates": [
                state_update[0].get_value()
                for state_update in self.srng.state_updates
            ]
        }

        np.savez(pjoin(savedir, "state.npz"), **state)
コード例 #6
0
    def save(self, path):
        savedir = smartutils.create_folder(pjoin(path, type(self).__name__))
        smartutils.save_dict_to_json_file(pjoin(savedir, "hyperparams.json"),
                                          self.hyperparameters)

        params = {param.name: param.get_value() for param in self.parameters}
        assert len(self.parameters) == len(
            params)  # Implies names are all unique.
        np.savez(pjoin(savedir, "params.npz"), **params)

        state = {
            "version":
            1,
            "_srng_rstate":
            self.srng.rstate,
            "_srng_state_updates": [
                state_update[0].get_value()
                for state_update in self.srng.state_updates
            ]
        }
        np.savez(pjoin(savedir, "state.npz"), **state)