Commit Graph

9 Commits

Author SHA1 Message Date
Arthur Meyre
0af91911c3 fix: accept other comparison functions for the new per cell checks in tests 2021-12-24 16:07:44 +01:00
jfrery
9e938ddfff feat: add missing pieces for relu6 2021-12-24 12:30:14 +01:00
Arthur Meyre
a0c26315ea chore: make check_is_good_execution a fixture and fix flaky tests using it
closes #1061
2021-12-03 17:43:11 +01:00
Arthur Meyre
500d46a552 chore: disable flaky test 2021-12-03 12:10:08 +01:00
Arthur Meyre
c7255dfd66 feat: update compile_torch_model to return compiled quantized module
closes #898
2021-12-03 10:32:04 +01:00
Benoit Chevallier-Mames
edefde9189 feat: adding a compile_torch_model function
refs #898
2021-11-30 16:24:21 +01:00
Benoit Chevallier-Mames
ec396effb2 chore: seed torch as much as possible
closes #877
2021-11-24 09:47:37 +01:00
jfrery
c5952cd09f feat: add quantization utilities 2021-11-10 18:25:31 +01:00
jfrery
fac7c9c954 feat: conversion from torch.nn.Module to numpy. 2021-11-09 11:02:48 +01:00