update readme

This commit is contained in:
Kohya S
2023-03-09 21:51:22 +09:00
parent aa317d4f57
commit cf2bc437ec
3 changed files with 38 additions and 36 deletions

View File

@@ -21,7 +21,7 @@ def main(file):
for key, value in values:
value = value.to(torch.float32)
print(f"{key},{torch.mean(torch.abs(value))},{torch.min(torch.abs(value))}")
print(f"{key},{str(tuple(value.size())).replace(', ', '-')},{torch.mean(torch.abs(value))},{torch.min(torch.abs(value))}")
if __name__ == '__main__':