Ng,dZGddZddZdS)zM Eval metrics and related Hacked together by / Copyright 2020 Ross Wightman c&eZdZdZdZdZddZdS) AverageMeterz1Computes and stores the average and current valuec.|dSN)resetselfs N/var/www/html/ai-engine/env/lib/python3.11/site-packages/timm/utils/metrics.py__init__zAverageMeter.__init__ s c>d|_d|_d|_d|_dS)N)valavgsumcountrs r rzAverageMeter.reset s" r c||_|xj||zz c_|xj|z c_|j|jz |_dSr)rrrr)rrns r updatezAverageMeter.updates@ C!G a 8dj(r Nr)__name__ __module__ __qualname____doc__r rrr r rrsL;; ))))))r rrctt||d|d|ddd\}}|}||dd|fd|DS)zNComputes the accuracy over the k top predictions for the specified values of krr Tcg|]X}dt|dddzz YS)Nrr gY@)minreshapefloatr).0k batch_sizecorrectmaxks r zaccuracy.. s] ` ` `VWGMSD\\M " * *2 . . 4 4 6 6 : :1 = = Dz Q ` ` `r )rmaxsizetopkteqr expand_as)outputtargetr*_predr$r%r&s @@@r accuracyr2s s4yy&++--* + +DQJkk$4..GAt 6688DggfnnQ++55d;;<r3s[ ))))))))$aaaaaar