MCPcopy Create free account
hub / github.com/ImprintLab/MedSegDiff / _optimize_fp16

Method _optimize_fp16

guided_diffusion/fp16_util.py:189–207  ·  view source on GitHub ↗
(self, opt: th.optim.Optimizer)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

optimizeMethod · 0.95

Calls 7

_compute_normsMethod · 0.95
check_overflowFunction · 0.85
zero_master_gradsFunction · 0.85
logkv_meanMethod · 0.80
logMethod · 0.80

Tested by

no test coverage detected