| 1234567891011121314151617 |
- ## @package dropout
- # Module caffe2.python.helpers.dropout
- def dropout(model, blob_in, blob_out, use_cudnn=False, **kwargs):
- """dropout"""
- if use_cudnn:
- kwargs['engine'] = 'CUDNN'
- else:
- kwargs['engine'] = 'DEFAULT'
- assert 'is_test' in kwargs, "Argument 'is_test' is required"
- return model.net.Dropout(
- blob_in, [blob_out, "_" + blob_out + "_mask"], **kwargs)[0]
|
