average_checkpoints(model_dir, output_dir, trackables, max_count=8, model_key='model')¶
Averages object-based checkpoints.
model_dir – The directory containing checkpoints.
output_dir – The directory that will contain the averaged checkpoint.
trackables – A dictionary containing the trackable objects included in the checkpoint.
max_count – The maximum number of checkpoints to average.
model_key – The key in
trackablesthat references the model.
The path to the directory containing the averaged checkpoint.
ValueError – if
output_diris the same as
ValueError – if a model is not found in
trackablesor is not already built.
ValueError – if no checkpoints are found in