Witllm/tools/test.py

25 lines
429 B
Python
Raw Normal View History

2023-12-21 19:52:19 +08:00
import show
import torch
2023-12-21 21:20:49 +08:00
# radata = torch.randn(8192, 128)
# show.DumpTensorToImage(radata, "test.png", autoPad=True,scale=0.2)
radata = torch.randn(127)
show.DumpTensorToImage(radata, "test.png")
2023-12-25 22:53:53 +08:00
2024-01-13 16:48:56 +08:00
radata = torch.randn(3,127,127)
show.DumpTensorToImage(radata, "test.png")
2023-12-25 22:53:53 +08:00
radata = torch.randn(127, 127)
show.DumpTensorToLog(radata, "test.log")
2023-12-29 19:55:53 +08:00
radata = torch.randn(127, 127) - 0.5
show.ProbGE0(radata)
show.DumpProb()