OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2025-12-06 12:20:27 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

typofix

Browse Source
This commit is contained in:
torzdf 2022-06-20 12:46:23 +01:00
parent 9e94273f89
commit ad408f07fb
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lib/gui/analysis/event_reader.py
View File

@ -365,10 +365,11 @@ class _Cache():
retval = {}
for idx, data in raw.items():
val = np.frombuffer(zlib.decompress(data[metric]), dtype=dtype)
buff = np.frombuffer(zlib.decompress(data[metric]), dtype=dtype)
shape = data[f"{metric}_shape"]
if len(shape) > 1:
val = val.reshape(-1, *shape[1:])
buff = buff.reshape(-1, *shape[1:])
val = {metric: buff}
if metric == "loss":
val["labels"] = data["labels"]
retval[idx] = val