diff options
Diffstat (limited to 'dnn/torch/rdovae/export_rdovae_weights.py')
-rw-r--r-- | dnn/torch/rdovae/export_rdovae_weights.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/dnn/torch/rdovae/export_rdovae_weights.py b/dnn/torch/rdovae/export_rdovae_weights.py index 8a6d0c03..f9c1db81 100644 --- a/dnn/torch/rdovae/export_rdovae_weights.py +++ b/dnn/torch/rdovae/export_rdovae_weights.py @@ -29,6 +29,9 @@ import os import argparse +import sys + +sys.path.append(os.path.join(os.path.dirname(__file__), '../weight-exchange')) parser = argparse.ArgumentParser() |