Name	Size	Mode	Actions
algebra.cpython-36.pyc	1503	0644	editdlrm
arg_scope.cpython-36.pyc	972	0644	editdlrm
array_helpers.cpython-36.pyc	686	0644	editdlrm
control_ops.cpython-36.pyc	649	0644	editdlrm
conv.cpython-36.pyc	5786	0644	editdlrm
db_input.cpython-36.pyc	451	0644	editdlrm
dropout.cpython-36.pyc	486	0644	editdlrm
elementwise_linear.cpython-36.pyc	1109	0644	editdlrm
fc.cpython-36.pyc	3492	0644	editdlrm
nonlinearity.cpython-36.pyc	1082	0644	editdlrm
normalization.cpython-36.pyc	5978	0644	editdlrm
pooling.cpython-36.pyc	903	0644	editdlrm
quantization.cpython-36.pyc	357	0644	editdlrm
tools.cpython-36.pyc	818	0644	editdlrm
train.cpython-36.pyc	1965	0644	editdlrm
__init__.cpython-36.pyc	157	0644	editdlrm

3 Eg��@sBdd
lm
Z
mZ
ddlmZ
ddd�
Zdd�Zdd �Zd d�ZdS) �)�core�scope)
� caffe2_pb2Ncs>�dkrtj
���d
kr&|jdd�S�f
dd�|jD�
SdS)N�c
sg|]}
|
j��kr|
�qS�)
ZGetNameScope)�.0�
w)
� namescoper�G/usr/local/lib64/python3.6/site-packages/caffe2/python/helpers/train.py� sz _get_weights..)rZCurrentNameScope�weights)�modelr r)
r r �_get_weightss


rcKsPd
|kr|d
=|jj
g|
fdg
dtjjtjtjd�d�|��

|jj |
|
f|�
S)N� device_option�
r
)�shape�valueZdtyper) �param_init_net�ConstantFillrZDataTypeZINT64�DeviceOptionr�CPU�netZIter)r �blob_out�kwargsrrr �iters









rcKs�d
|kr|d
ntj
�}|dkp*|jtjk}|r�d|kr�|ddkr�|jj|
d|
dd�}|jj|
d|
dd�}|jj||g|fd
tj tjd�i
|��

n|jj|
|�
dS)NrZtop_krr
Z_host) rZCurrentDeviceScopeZdevice_typerrrZCopyGPUToCPUZAccuracyrr)r Zblob_inrr�devZis_cpuZ pred_hostZ label_hostrrr �accuracy%s








rcCsn|
d
k
rdS|jj
gddg
|
d�}|jj
gddg
dd�}x0t|�
D]$}|j|}|jj||||g|�
qBWdS)z�Adds a decay to weights in the model. This is a form of L2 regularization. Args: weight_decay: strength of the regularization gN�wdr)rr�ONEg�?)rrrZ param_to_gradrZWeightedSum)r Zweight_decayrr�paramZgradrrr �add_weight_decay:s






r )
N) Z caffe2.pythonrrZcaffe2.protorrrrr rrrr �s