File size: 400 Bytes
74d2eb9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import torch
def model_information(path):
model_data = torch.load(path, map_location="cpu")
print(f"Loaded model from {path}")
data = model_data
epochs = data.get("info", "None")
sr = data.get("sr", "None")
f0 = data.get("f0", "None")
version = data.get("version", "None")
return(f"Epochs: {epochs}\nSampling rate: {sr}\nPitch guidance: {f0}\nVersion: {version}") |