import os import torch def change_info(path, info, name): try: ckpt = torch.load(path, map_location="cpu") ckpt["info"] = info if name == "": name = os.path.basename(path) torch.save(ckpt, f"logs/{name}/{name}") return "Success." except Exception as error: print(error)