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 |
|