def load(self, path: str = None):
loaded_items = util.load_pytree(path)
self.params = loaded_items["params"]
self.state = loaded_items["state"]
self.opt_state = loaded_items["opt_state"]
self.train_losses = loaded_items["train_losses"]
self.test_losses = loaded_items["test_losses"]
def load(self, path: str=None):
loaded_items = util.load_pytree(path)
self.params = loaded_items["params"]
self.state = loaded_items["state"]
self.test_eval_times = loaded_items["test_eval_times"]
self.trainer.load_items(loaded_items)
self.tester.load_items(loaded_items)
def load(self, path: str=None): loaded_items = util.load_pytree(path) self.params = loaded_items["params"] self.state = loaded_items["state"]