File size: 550 Bytes
13e4f50 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | import torch
from safetensors.torch import load_file
def load_model(path='model.safetensors'):
return load_file(path)
def atmost5(bits, weights):
"""At-most-5-out-of-8: fires when 5 or fewer inputs are active."""
inp = torch.tensor([float(b) for b in bits])
return int((inp * weights['weight']).sum() + weights['bias'] >= 0)
if __name__ == '__main__':
w = load_model()
print('AtMost5OutOf8: HW <= 5')
for hw in range(9):
bits = [1]*hw + [0]*(8-hw)
print(f'HW={hw}: {atmost5(bits, w)}')
|