MCPcopy Create free account
hub / github.com/HA-Transformer/MAT / average_checkpoints

Function average_checkpoints

scripts/average_checkpoints.py:14–67  ·  view source on GitHub ↗

Loads checkpoints from inputs and returns a model with averaged weights. Args: inputs: An iterable of string paths of checkpoints to load from. Returns: A dict of string keys mapping to various values. The 'model' key from the returned dict should correspond to an Ordered

(inputs)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 3

mainFunction · 0.85

Calls 1

loadMethod · 0.80