deepmd.tf.entrypoints.transfer

Module used for transferring parameters between models.

Functions

transfer(→ None)

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

Module Contents

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

Transfer operation from old from 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 transferred parameters will be saved to this location

**kwargs

additional arguments