diff options
Diffstat (limited to 'dnn/torch/osce')
-rw-r--r-- | dnn/torch/osce/export_model_weights.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/dnn/torch/osce/export_model_weights.py b/dnn/torch/osce/export_model_weights.py index da90a39a..3aac3a0c 100644 --- a/dnn/torch/osce/export_model_weights.py +++ b/dnn/torch/osce/export_model_weights.py @@ -60,13 +60,13 @@ schedules = { ('feature_net.conv1', dict()), ('feature_net.conv2', dict(quantize=True, scale=None)), ('feature_net.tconv', dict(quantize=True, scale=None)), - ('feature_net.gru', dict()), + ('feature_net.gru', dict(quantize=True, scale=None, recurrent_scale=None)), ('cf1', dict(quantize=True, scale=None)), ('cf2', dict(quantize=True, scale=None)), ('af1', dict(quantize=True, scale=None)), - ('tdshape1', dict()), - ('tdshape2', dict()), - ('tdshape3', dict()), + ('tdshape1', dict(quantize=True, scale=None)), + ('tdshape2', dict(quantize=True, scale=None)), + ('tdshape3', dict(quantize=True, scale=None)), ('af2', dict(quantize=True, scale=None)), ('af3', dict(quantize=True, scale=None)), ('af4', dict(quantize=True, scale=None)), @@ -81,7 +81,7 @@ schedules = { ('feature_net.conv1', dict()), ('feature_net.conv2', dict(quantize=True, scale=None)), ('feature_net.tconv', dict(quantize=True, scale=None)), - ('feature_net.gru', dict()), + ('feature_net.gru', dict(quantize=True, scale=None, recurrent_scale=None)), ('cf1', dict(quantize=True, scale=None)), ('cf2', dict(quantize=True, scale=None)), ('af1', dict(quantize=True, scale=None)) |