未验证 提交 8053f3f4 编写于 作者: 隔壁的泰山233 提交者: GitHub

Change default dropout rate.

上级 9e807c81
......@@ -92,7 +92,7 @@ class ResidualBlock(Module):
Each resolution is processed with two residual blocks.
"""
def __init__(self, in_channels: int, out_channels: int, time_channels: int, n_groups: int = 32, dropout_rate: float = 0.01):
def __init__(self, in_channels: int, out_channels: int, time_channels: int, n_groups: int = 32, dropout_rate: float = 0.1):
"""
* `in_channels` is the number of input channels
* `out_channels` is the number of input channels
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册