Change default dropout rate.

Author: jakehsiao

labml_nn/diffusion/ddpm/unet.py

@@ -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