diff options
| author | ericmarin <maarin.eric@gmail.com> | 2026-03-21 11:47:40 +0100 |
|---|---|---|
| committer | ericmarin <maarin.eric@gmail.com> | 2026-03-21 12:00:16 +0100 |
| commit | e2abe9d9ec649b849cc39b516c1db1b4fa592003 (patch) | |
| tree | d74dcc2e0691bb587d2a9a695639517d3aec9256 /xor/xor.py | |
| parent | af4335cf47984576e7493a0eb6569d3f6ecc31c8 (diff) | |
| download | vein-e2abe9d9ec649b849cc39b516c1db1b4fa592003.tar.gz vein-e2abe9d9ec649b849cc39b516c1db1b4fa592003.zip | |
created class
Diffstat (limited to '')
| -rw-r--r-- | xor/xor.py (renamed from xor.py) | 15 |
1 files changed, 2 insertions, 13 deletions
@@ -1,7 +1,6 @@ import torch import torch.nn as nn import torch.onnx -import nneq class xor_mlp(nn.Module): def __init__(self, hidden_dim): @@ -37,15 +36,5 @@ if __name__ == "__main__": torch_net_a = train_model("Network A", 8).eval() torch_net_b = train_model("Network B", 16).eval() - onnx_net_a = torch.onnx.export(torch_net_a, (torch.randn(1, 2),), verbose=False).model_proto # type: ignore - onnx_net_b = torch.onnx.export(torch_net_b, (torch.randn(1, 2),), verbose=False).model_proto # type: ignore - - z3_net_a = nneq.net(onnx_net_a) - z3_net_b = nneq.net(onnx_net_b) - - print("") - nneq.strict_equivalence(z3_net_a, z3_net_b) - print("") - nneq.epsilon_equivalence(z3_net_a, z3_net_b, 0.1) - print("") - nneq.argmax_equivalence(z3_net_a, z3_net_b) + torch.onnx.export(torch_net_a, (torch.randn(1, 2),), "xor_a.onnx") + torch.onnx.export(torch_net_b, (torch.randn(1, 2),), "xor_b.onnx") |