U
    d}                     @   s   d d Z dS )c                 C   s.   d| }| j jg |||d}| jj|||dS )NZ	dbreader_)dbdb_type)
batch_size)Zparam_init_netZCreateDBnetZTensorProtosDBInput)modelZ	blobs_outr   r   r   Zdbreader_nameZdbreader r   B/tmp/pip-unpacked-wheel-ua33x9lu/caffe2/python/helpers/db_input.pydb_input   s      r   N)r   r   r   r   r   <module>       