o xi@s`ddlZddlZddlmZmZddZGdddejdZGd d d eZ Gd d d eZ dS) N) SummaryItem SummaryRecordcCst|tr|St|SN) isinstancedictvars)dr K/home/ubuntu/.local/lib/python3.10/site-packages/wandb/sdk/wandb_summary.py _get_dicts r c@sveZdZdZejddZejdefddZddZ dd d Z d d Z e Z ddZ e ZddZeZdefddZd S) SummaryDictzdict-like wrapper for the nested dictionaries in a SummarySubDict. Triggers self._root._callback on property changes. cCtrNotImplementedErrorselfr r r _as_dictzSummaryDict._as_dictrecordcCrrrrrr r r _updaterzSummaryDict._updatecCsdd|DS)NcSsg|]}|dkr|qS)_wandbr ).0kr r r sz$SummaryDict.keys..)rrr r r keyszSummaryDict.keysNcCs|||Sr)rget)rkeydefaultr r r rszSummaryDict.getcCsN||}t|tr%t}t|d|t|d|t|d||S|SN_items_parent _parent_key)rrrSummarySubDictobject __setattr__)rritem wrapped_itemr r r __getitem__"s  zSummaryDict.__getitem__cCs|||idSr)update)rrvalr r r __setitem__3rzSummaryDict.__setitem__cCs*t}t}|f|_|f|_||dSr)rrrremover)rrrr(r r r __delattr__8s zSummaryDict.__delattr__r cCsFt}|D]\}}t}|f|_||_|j|q||dSr)ritemsrrvaluer+appendr)rr rrr1r(r r r r+AszSummaryDict.updater)__name__ __module__ __qualname____doc__abcabstractmethodrrrrrr* __getattr__r-r'r/ __delitem__rr+r r r r r s  r ) metaclasscsdeZdZUdZejed<ejed<dejffdd Zdejfdd Zd d Z d e fd dZ Z S)Summarya3Track single values for each metric for each run. By default, a metric's summary is the last value of its History. For example, `wandb.log({'accuracy': 0.9})` will add a new step to History and update Summary to the latest value. In some cases, it's more useful to have the maximum or minimum of a metric instead of the final value. You can set history manually `(wandb.summary['accuracy'] = best_acc)`. In the UI, summary metrics appear in the table to compare across runs. Summary metrics are also used in visualizations like the scatter plot and parallel coordinates chart. After training has completed, you may want to save evaluation metrics to a run. Summary can handle numpy arrays and PyTorch/TensorFlow tensors. When you save one of these types to Summary, we persist the entire tensor in a binary file and store high level metrics in the summary object, such as min, mean, variance, and 95th percentile. Examples: ```python wandb.init(config=args) best_accuracy = 0 for epoch in range(1, args.epochs + 1): test_loss, test_accuracy = test() if test_accuracy > best_accuracy: wandb.run.summary["best_accuracy"] = test_accuracy best_accuracy = test_accuracy ``` _update_callback_get_current_summary_callbackget_current_summary_callbackcs*tt|ddt|d|dS)Nr=r>)super__init__r&r')rr? __class__r r rAps zSummary.__init__update_callbackcCst|d|dS)Nr=)r&r')rrDr r r _set_update_callbackwrzSummary._set_update_callbackcCs|Sr)r>rr r r rzszSummary._as_dictrcCs|jr ||dSdSr)r=rr r r r}szSummary._update) r3r4r5r6tCallable__annotations__rArErrr __classcell__r r rBr r<Ls  r<c@sHeZdZUdZeed<eed<eed<ddZddZ d e fd d Z d S) r%z_Non-root node of the summary data structure. Contains a path to itself from the root. r"r#r$cCs0t|dtt|ddt|dddSr!)r&r'rrr r r rAszSummarySubDict.__init__cCs|jSr)r"rr r r rszSummarySubDict._as_dictrcCs|j||jSr)r#r_add_next_parentr$rr r r rszSummarySubDict._updateN) r3r4r5r6rrHr strrArrrr r r r r%s r%) r7typingrFinterface.summary_recordrrr ABCMetar r<r%r r r r s>6