MCPcopy Create free account
hub / github.com/IronSpiderMan/MuseTalkPlus / training_loop

Function training_loop

scripts/train_musetalk.py:33–126  ·  view source on GitHub ↗
(
        models, train_loader, test_loader, optimizer, epochs, accelerator, gamma=2.0,
        max_grad_norm=1, output_dir="outputs"
)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

mainFunction · 0.70

Calls 2

save_modelFunction · 0.90
evaluateFunction · 0.70

Tested by

no test coverage detected