From 87fa48b91c0f33a0ac396ef8524965ce74c6f95d Mon Sep 17 00:00:00 2001 From: Jinhyuk Lee Date: Tue, 18 Jan 2022 18:18:24 +0900 Subject: [PATCH] minor --- densephrases/utils/embed_utils.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/densephrases/utils/embed_utils.py b/densephrases/utils/embed_utils.py index 680223d..eed062a 100644 --- a/densephrases/utils/embed_utils.py +++ b/densephrases/utils/embed_utils.py @@ -14,8 +14,6 @@ import h5py import numpy as np import torch -import sys -import pdb from functools import partial from multiprocessing import Queue, Process, Pool, cpu_count @@ -139,6 +137,7 @@ def float_to_int8(num, offset, factor): def int8_to_float(num, offset, factor): return num.astype(np.float32) / factor + offset + def compress_metadata(metadata, dense_offset, dense_scale, record_histogram=False): for key in ['start']: if key in metadata: