theta_0[key]=theta_func(theta_0[key],theta_1[key],(float(1.0)-interp_amount))# Need to reverse the interp_amount to match the desired mix ration in the merged checkpoint
forkeyintheta_1.keys():
forkeyintheta_1.keys():
if'model'inkeyandkeynotintheta_0:
if'model'inkeyandkeynotintheta_0:
...
@@ -188,7 +188,7 @@ def run_modelmerger(primary_model_name, secondary_model_name, interp_method, int
...
@@ -188,7 +188,7 @@ def run_modelmerger(primary_model_name, secondary_model_name, interp_method, int