8 lines
229 B
Python
8 lines
229 B
Python
import glob
|
|
|
|
import torch
|
|
|
|
|
|
for path in sorted(glob.glob("/artifacts/fl2va-sampler-reference/*.pt")):
|
|
state = torch.load(path, map_location="cpu", weights_only=False)
|
|
print(path.rsplit("/", 1)[-1], float(state["sigma"]))
|