B
|
_( ã @ s. d Z ddlZddlmZ G dd dejZdS )z9
|
@author: liaoxingyu
|
@contact: sherlockliao01@gmail.com
|
é N)Únnc s( e Zd ZdZ fddZdd Z ZS )Ú BatchDropzrref: https://github.com/daizuozhuo/batch-dropblock-network/blob/master/models/networks.py
|
batch drop mask
|
c s t t| ¡ || _|| _d S )N)Úsuperr Ú__init__Úh_ratioÚw_ratio)Úselfr r )Ú __class__© ú./layers/batch_drop.pyr s zBatchDrop.__init__c C s | j r| ¡ dd
\}}t| j| }t| j| }t d|| ¡}t d|| ¡}| | ¡ ¡}d|d d
d d
|||
|||
f< || }|S )Néþÿÿÿr )ÚtrainingÚsizeÚroundr r ÚrandomÚrandintÚnew_ones) r ÚxÚhÚwZrhÚrwÚsxZsyÚmaskr
|
r
|
r Úforward s (zBatchDrop.forward)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r Ú __classcell__r
|
r
|
)r r r s r )r r Útorchr ÚModuler r
|
r
|
r
|
r Ú<module> s
|
