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)
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)
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)
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)
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)
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)