-
Notifications
You must be signed in to change notification settings - Fork 133
Different ways to import parameters
Albert Zeyer edited this page Sep 18, 2021
·
19 revisions
There are multiple ways how parameters can be imported from some other model or checkpoint.
This is the most flexible option.
See the scripts tf_avg_checkpoints.py
or tf_inspect_checkpoint.py
as examples.
It should be straightforward to write some own custom logic.
The parameters must match exactly. In a new training (first epoch), instead of random initialization, it would load the given model checkpoint.
The param init uses get_initializer
and can in principle use any initializer (e.g. load_txt_file_initializer
), or even custom initializing code, which could import other parameters from a checkpoint or elsewhere.