прочитать модель, изменить вес первого конвон слоя, затем сохранить измененную модель

Question

model_dir = './pb/model.ckpt'

reader = pywrap_tensorflow.NewCheckpointReader(model_dir)
param_dict = reader.get_variable_to_shape_map()

for key,val in param_dict.items():
    if key == 'FeatureExtractor/MobilenetV2/Conv/weights':

        // here I want to change this weights to some other values
// here I want to save the changed model