mirror of
https://github.com/marian-nmt/marian.git
synced 2024-12-11 09:54:22 +03:00
31 lines
803 B
Python
31 lines
803 B
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
import os, sys
|
||
|
|
import argparse
|
||
|
|
import numpy as np;
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
parser.add_argument('-d', '--dropout', type=float, required=True,
|
||
|
|
help="dropout rate")
|
||
|
|
parser.add_argument('-i', '--input', required=True,
|
||
|
|
help="Input model")
|
||
|
|
parser.add_argument('-o', '--output', required=True,
|
||
|
|
help="Output model")
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
|
||
|
|
multiplier = 1.0 - args.dropout
|
||
|
|
|
||
|
|
output = dict()
|
||
|
|
print "Loading", args.input, "to multiple with", multiplier
|
||
|
|
with open(args.input, "rb") as mfile:
|
||
|
|
m = np.load(mfile)
|
||
|
|
for k in m:
|
||
|
|
if "history_errs" in k or "_b" in k or "c_tt" in k:
|
||
|
|
output[k] = m[k]
|
||
|
|
else:
|
||
|
|
output[k] = multiplier * m[k]
|
||
|
|
|
||
|
|
print "Saving to", args.output
|
||
|
|
np.savez(args.output, **output)
|