MCPcopy Index your code
hub / github.com/openai/improved-diffusion / _setup_fp16

Method _setup_fp16

improved_diffusion/train_util.py:157–159  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

155 self.opt.load_state_dict(state_dict)
156
157 def _setup_fp16(self):
158 self.master_params = make_master_params(self.model_params)
159 self.model.convert_to_fp16()
160
161 def run_loop(self):
162 while (

Callers 1

__init__Method · 0.95

Calls 2

make_master_paramsFunction · 0.85
convert_to_fp16Method · 0.80

Tested by

no test coverage detected