def load_cache(self, args, path, epoch):
    path = self.get_cache_path_load(args, path, epoch)
    assert (path is not None), 'no cache file'
    dt = torch.load(path)
    self.epoch = dt['epoch']
    self.base_lr = dt['base_lr']
    self.model.load_state_dict(dt['model'])
def load_cache(self, args, path, epoch):
    path = self.get_cache_path_load(args, path, epoch)
    assert (path is not None), 'no cache file'
    dt = torch.load(path)
    self.epoch = dt['epoch']
    self.base_lr = dt['base_lr']
    self.model.load_state_dict(dt['model'])