MCPcopy Index your code
hub / github.com/Turing-Project/WriteGPT / _tally_parameters

Function _tally_parameters

LanguageNetwork/BERT/models/trainer.py:16–18  ·  view source on GitHub ↗
(model)

Source from the content-addressed store, hash-verified

14
15
16def _tally_parameters(model):
17 n_params = sum([p.nelement() for p in model.parameters()])
18 return n_params
19
20
21def build_trainer(args, device_id, model, optim):

Callers 1

build_trainerFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected