Skip to content

GitLab

"megatron/git@developer.sourcefind.cn:OpenDAS/megatron-lm.git" did not exist on "adec01d055ed4fa335cf7432ddb93f65fc9f9829"
Commit 5d981d57 authored by ScottSWu's avatar ScottSWu
Browse files

Fix division changing dtype to float in python3

parent afdcf7d4
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
compression/encoder.py
View file @ 5d981d57
...@@ -92,7 +92,7 @@ def main(_): ...@@ -92,7 +92,7 @@ def main(_):
int_codes = np.asarray([x.astype(np.int8) for x in results]) int_codes = np.asarray([x.astype(np.int8) for x in results])
# Convert int codes to binary. # Convert int codes to binary.
int_codes = (int_codes + 1)/2 int_codes = (int_codes + 1)//2
export = np.packbits(int_codes.reshape(-1)) export = np.packbits(int_codes.reshape(-1))
output = io.BytesIO() output = io.BytesIO()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment