15 lines
340 B
Python
15 lines
340 B
Python
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)
|