mirror of
https://github.com/marian-nmt/marian.git
synced 2024-09-17 09:47:34 +03:00
17 lines
378 B
Python
Executable File
17 lines
378 B
Python
Executable File
import numpy as np
|
|
import sys
|
|
|
|
d = dict()
|
|
m = np.load(sys.argv[1])
|
|
for k in m:
|
|
if "ff_" == k[0:3]:
|
|
d["decoder_" + k] = m[k]
|
|
elif k == "special:model.yml":
|
|
info = m[k].tobytes()
|
|
info = info.replace("layers-dec", "dec-depth")
|
|
info = info.replace("layers-enc", "enc-depth")
|
|
d[k] = info
|
|
print info
|
|
else:
|
|
d[k] = m[k]
|
|
np.savez(sys.argv[1] + ".fixed", **d) |