o 㥵ih@s2ddlmZddlmZededdfddZdS))rank_zero_only)logger object_dictreturnNcCsi}|d}|d}|d}|jstddS|d|d<tdd|D|d<td d|D|d <td d|D|d <|d |d <|d|d<|d|d<|d|d<|d|d<|d|d<|d|d<|d|d<|jD]}||q~dS)zzControls which config parts are saved by lightning loggers. Additionally saves: - Number of model parameters cfgmodeltrainerz4Logger not found! Skipping hyperparameter logging...Ncss|]}|VqdSN)numel.0prS/home/ubuntu/.local/lib/python3.10/site-packages/fish_speech/utils/logging_utils.py sz&log_hyperparameters..zmodel/params/totalcss|] }|jr|VqdSr  requires_gradr r rrrr  zmodel/params/trainablecss|] }|js|VqdSr rr rrrrrzmodel/params/non_trainabledata callbacksextras task_nametags ckpt_pathseed)rlogwarningsum parametersgetloggerslog_hyperparams)rhparamsrrrrrrrlog_hyperparameterss4       r#)lightning.pytorch.utilitiesrfish_speech.utilsrrdictr#rrrrs