Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 20:49:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update doc for fp16util.py (#50)

Browse Source 
update url, original url is not found
This commit is contained in:
chenxujun
2022-12-15 10:27:44 +08:00
committed by GitHub
parent 664727cf16
commit e6961b3970
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/utils/multi_modal/fp16/fp16util.py
View File

@@ -123,7 +123,7 @@ def prep_param_lists(model, flat_master=False):
Currently, if ``flat_master=True``, all the model's parameters must be the same type. If the model has parameters of different types, use ``flat_master=False``, or use :class:`FP16_Optimizer`. # noqa
.. _`Training Neural Networks with Mixed Precision: Real Examples`:
http://on-demand.gputechconf.com/gtc/2018/video/S81012/
https://www.nvidia.com/en-us/on-demand/session/gtcsiliconvalley2018-s81012/
"""
model_params = [
param for param in model.parameters() if param.requires_grad