deepmd.tf.entrypoints.transfer

Module used for transfering parameters between models.

Module Contents

Functions

transfer(*, old_model, raw_model, output, **kwargs)

Transfer operation from old fron graph to new prepared raw graph.

deepmd.tf.entrypoints.transfer.transfer(*, old_model: str, raw_model: str, output: str, **kwargs)

Transfer operation from old fron graph to new prepared raw graph.

Parameters:

old_modelstr

frozen old graph model

raw_modelstr

new model that will accept ops from old model

outputstr

new model with transfered parameters will be saved to this location

**kwargs

additional arguments