a ==ic@sBddlmZmZddlmZd ddZddZdd Zd d ZdS) )corescope) caffe2_pb2Ncs>durtdkr&|jddSfdd|jDSdS)Ncsg|]}|kr|qS)Z GetNameScope).0w namescoperl/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/caffe2/python/helpers/train.py z _get_weights..)rZCurrentNameScopeweights)modelr rr r _get_weights s rc KsTd|vr|d=|jjg|fdgdtjjttjdd||jj ||fi|S)N device_optionr)shapevalueZdtyper) param_init_net ConstantFillrZDataTypeZINT64 DeviceOptionrCPUnetZIter)rblob_outkwargsrrr iters  rcKsd|vr|dnt}|dup*|jtjk}|sd|vr|ddkr|j|d|dd}|j|d|dd}|jj||g|fdt tjdi|n|j||dS)NrZtop_krrZ_host) rZCurrentDeviceScopeZ device_typerrrZ CopyGPUToCPUZAccuracyrr)rZblob_inrrdevZis_cpuZ pred_hostZ label_hostrrr accuracy%s  rcCsj|dkr dS|jjgddg|d}|jjgddgdd}t|D]$}|j|}|j||||g|q@dS)zAdds a decay to weights in the model. This is a form of L2 regularization. Args: weight_decay: strength of the regularization gNwdr)rrONEg?)rrrZ param_to_gradrZ WeightedSum)rZ weight_decayrr paramZgradrrr add_weight_decay:s    r")N) Z caffe2.pythonrrZ caffe2.protorrrrr"rrrr s