diff --git a/benchmark/torchscripts/BetaCoOccurringFD.pt b/benchmark/torchscripts/BetaCoOccurringFD.pt new file mode 100644 index 00000000..aee2bf21 Binary files /dev/null and b/benchmark/torchscripts/BetaCoOccurringFD.pt differ diff --git a/benchmark/torchscripts/beta-CoOccurringFD.py b/benchmark/torchscripts/BetaCoOccurringFD.py similarity index 94% rename from benchmark/torchscripts/beta-CoOccurringFD.py rename to benchmark/torchscripts/BetaCoOccurringFD.py index 646bc67c..9b581b83 100644 --- a/benchmark/torchscripts/beta-CoOccurringFD.py +++ b/benchmark/torchscripts/BetaCoOccurringFD.py @@ -17,9 +17,9 @@ def attenuate(beta: float, k: torch.Tensor, l: int) -> torch.Tensor: @torch.jit.script -def FDAMM(A: torch.Tensor, B: torch.Tensor, l: int, beta: float): +def FDAMM(A: torch.Tensor, B: torch.Tensor, l: int): B = B.t() - + beta=28.0 assert A.shape[1] == B.shape[1] mx, n = A.shape my, n = B.shape @@ -91,7 +91,7 @@ def main(): t = time.time() - aResult = FDAMM(A, B, 500, 28.0) + aResult = FDAMM(A, B, 500) print("approximate: " + str(time.time() - t) + "s") print(aResult) @@ -105,7 +105,7 @@ def main(): print("\nerror: " + str(torch.norm(aResult - eResult, p='fro').item())) - FDAMM_script = FDAMM.save("beta-Co-Occurring FD.pt") + FDAMM_script = FDAMM.save("BetaCoOccurringFD.pt") if __name__ == '__main__': main() diff --git a/benchmark/torchscripts/CoOccurringFD.pt b/benchmark/torchscripts/CoOccurringFD.pt index 8f75adea..6765f9cc 100644 Binary files a/benchmark/torchscripts/CoOccurringFD.pt and b/benchmark/torchscripts/CoOccurringFD.pt differ diff --git a/benchmark/torchscripts/CoOccurringFD.py b/benchmark/torchscripts/CoOccurringFD.py index 7272058f..688da6df 100644 --- a/benchmark/torchscripts/CoOccurringFD.py +++ b/benchmark/torchscripts/CoOccurringFD.py @@ -108,7 +108,7 @@ def main(): print("\nerror: " + str(torch.norm(aResult - eResult, p='fro').item())) - FDAMM_script = FDAMM.save("Co-Occurring FD.pt") + FDAMM_script = FDAMM.save("CoOccurringFD.pt") if __name__ == '__main__': main() diff --git a/benchmark/torchscripts/beta-CoOccurringFD.pt b/benchmark/torchscripts/beta-CoOccurringFD.pt deleted file mode 100644 index 80048d96..00000000 Binary files a/benchmark/torchscripts/beta-CoOccurringFD.pt and /dev/null differ diff --git a/commit.sh b/commit.sh index 8a16d378..6ce14ccd 100755 --- a/commit.sh +++ b/commit.sh @@ -1,4 +1,4 @@ -BRANCH=main +BRANCH=hump git init git checkout -b $BRANCH git add .