Spaces:
Running
on
Zero
Running
on
Zero
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) | |