search

openai / consistency_models

Official repo for consistency models.
MIT License
6.02k stars 409 forks source link

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

Closed KeiKinn closed 7 hours ago

KeiKinn commented 1 year ago
  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.