Commit a8dc4de5 authored by Xiaohui Zhang's avatar Xiaohui Zhang Committed by Facebook GitHub Bot
Browse files

fix doc of specaugment transform (#3314)

Summary: Pull Request resolved: https://github.com/pytorch/audio/pull/3314

Reviewed By: nateanl

Differential Revision: D45621958

Pulled By: xiaohui-zhang

fbshipit-source-id: 17555a865790adadc2abd40a86571596386a12fc
parent 05ef7dc6
......@@ -1315,7 +1315,7 @@ class SpecAugment(torch.nn.Module):
This option is applicable only when the input tensor is 4D. (Default: ``True``)
p (float, optional): maximum proportion of time steps that can be masked.
Must be within range [0.0, 1.0]. (Default: 1.0)
zero_masking (bool, optional): If ``True``, use 0 as the mask value with 0,
zero_masking (bool, optional): If ``True``, use 0 as the mask value,
else use mean of the input tensor. (Default: ``False``)
"""
__constants__ = [
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment