consistency_models icon indicating copy to clipboard operation
consistency_models copied to clipboard
verified publisher icon openai

Fix bug when mixed precision training is set to off by use_fp16 == false

Open KeiKinn opened this issue 3 years ago • 0 comments

  1. The original code in train_utils.py only supports for fp16 precision.
  2. target_model_master_params will be set as list(target_model.parameters()) if use_fp16 == false.

KeiKinn avatar Apr 18 '23 22:04 KeiKinn